Eddie Pick commited on
Commit
9438062
·
unverified ·
1 Parent(s): c7143b1

Added OpenRouter

Browse files
Files changed (1) hide show
  1. models.py +5 -1
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(model_name=model, temperature=temperature)
 
 
 
 
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'