Spaces:
Running
Running
File size: 3,962 Bytes
bc6d235 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | import os
import requests
import sys
from pathlib import Path
# Tentativa 1: python-dotenv
try:
from dotenv import load_dotenv
loaded = load_dotenv()
print(f"INFO: python-dotenv carregou .env? {'Sim' if loaded else 'Não (arquivo não encontrado ou erro)'}")
except ImportError:
print("INFO: python-dotenv não instalado. Vou tentar ler .env manualmente.")
# Tentativa 2: Carregamento Manual (Fallback)
def manual_load_env():
env_path = Path(".env")
if env_path.exists():
print(f"INFO: Carregando {env_path.absolute()} manualmente...")
with open(env_path, "r", encoding="utf-8") as f:
for line in f:
line = line.strip()
if line and not line.startswith("#") and "=" in line:
key, value = line.split("=", 1)
os.environ[key.strip()] = value.strip()
return True
return False
if not os.getenv("MISTRAL_API_KEY"):
manual_load_env()
print(f"CWD: {os.getcwd()}")
print(f"Arquivos no CWD: {os.listdir('.')}")
def test_mistral():
print("\n--- Testando Mistral ---")
key = os.getenv("MISTRAL_API_KEY", "").strip()
if not key:
print("❌ MISTRAL_API_KEY não encontrada no ambiente.")
return
if (key.startswith('"') and key.endswith('"')) or (key.startswith("'") and key.endswith("'")):
key = key[1:-1]
print(f"Chave encontrada (prefixo): {key[:6]}...")
url = "https://api.mistral.ai/v1/chat/completions"
headers = {"Authorization": f"Bearer {key}", "Content-Type": "application/json"}
payload = {
"model": "mistral-large-latest",
"messages": [{"role": "user", "content": "Oi"}],
"max_tokens": 10
}
try:
response = requests.post(url, headers=headers, json=payload, timeout=10)
if response.status_code == 200:
print("✅ Mistral OK!")
else:
print(f"❌ Mistral erro {response.status_code}: {response.text}")
except Exception as e:
print(f"💥 Erro na requisição Mistral: {e}")
def test_gemini():
print("\n--- Testando Gemini ---")
key = os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY", "").strip()
if not key:
print("❌ Chave Gemini/Google não encontrada.")
return
print(f"Chave encontrada (prefixo): {key[:6]}...")
# Teste via endpoint v1 estável
url = f"https://generativelanguage.googleapis.com/v1/models/gemini-2.0-flash:generateContent?key={key}"
payload = {"contents": [{"parts":[{"text": "Oi"}]}]}
try:
response = requests.post(url, json=payload, timeout=10)
if response.status_code == 200:
print("✅ Gemini OK!")
else:
print(f"❌ Gemini erro {response.status_code}: {response.text}")
except Exception as e:
print(f"💥 Erro na requisição Gemini: {e}")
def test_groq():
print("\n--- Testando Groq ---")
key = os.getenv("GROQ_API_KEY", "").strip()
if not key:
print("❌ GROQ_API_KEY não encontrada.")
return
url = "https://api.groq.com/openai/v1/chat/completions"
headers = {"Authorization": f"Bearer {key}", "Content-Type": "application/json"}
payload = {
"model": "llama-3.3-70b-versatile",
"messages": [{"role": "user", "content": "Oi"}],
"max_tokens": 10
}
try:
response = requests.post(url, headers=headers, json=payload, timeout=10)
if response.status_code == 200:
print("✅ Groq OK!")
else:
print(f"❌ Groq erro {response.status_code}: {response.text}")
except Exception as e:
print(f"💥 Erro na requisição Groq: {e}")
if __name__ == "__main__":
print(f"Python: {sys.version}")
test_mistral()
test_gemini()
test_groq()
|