Spaces:
Running
Running
Eddie Pick
commited on
Added OpenRouter
Browse files
models.py
CHANGED
@@ -65,7 +65,11 @@ def get_model(provider_model, temperature=0.0):
|
|
65 |
case 'openai':
|
66 |
if model is None:
|
67 |
model = "gpt-4o-mini"
|
68 |
-
chat_llm = ChatOpenAI(
|
|
|
|
|
|
|
|
|
69 |
case 'perplexity':
|
70 |
if model is None:
|
71 |
model = 'llama-3.1-sonar-small-128k-online'
|
|
|
65 |
case 'openai':
|
66 |
if model is None:
|
67 |
model = "gpt-4o-mini"
|
68 |
+
chat_llm = ChatOpenAI(model=model, temperature=temperature)
|
69 |
+
case 'openrouter':
|
70 |
+
if model is None:
|
71 |
+
model = "google/gemini-flash-1.5-exp"
|
72 |
+
chat_llm = ChatOpenAI(model=model, temperature=temperature, base_url="https://openrouter.ai/api/v1", api_key=os.getenv("OPENROUTER_API_KEY"))
|
73 |
case 'perplexity':
|
74 |
if model is None:
|
75 |
model = 'llama-3.1-sonar-small-128k-online'
|