pvanand commited on
Commit
6bfafac
·
verified ·
1 Parent(s): c6cf4a7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +8 -5
main.py CHANGED
@@ -88,13 +88,16 @@ def update_db(user_id, conversation_id, message, response):
88
 
89
  ModelID = Literal[
90
  "openai/gpt-4o-mini",
91
- "meta-llama/llama-3-70b-instruct",
 
92
  "anthropic/claude-3.5-sonnet",
93
- "deepseek/deepseek-coder",
94
  "anthropic/claude-3-haiku",
95
- "openai/gpt-3.5-turbo-instruct",
96
- "qwen/qwen-72b-chat",
97
- "google/gemma-2-27b-it"
 
 
 
98
  ]
99
  class FollowupQueryModel(BaseModel):
100
  query: str = Field(..., description="User's query for the followup agent")
 
88
 
89
  ModelID = Literal[
90
  "openai/gpt-4o-mini",
91
+ "openai/chatgpt-4o-latest",
92
+ "deepseek/deepseek-chat",
93
  "anthropic/claude-3.5-sonnet",
 
94
  "anthropic/claude-3-haiku",
95
+ "meta-llama/llama-3.3-70b-instruct",
96
+ "google/gemini-2.0-flash-exp:free",
97
+ "google/gemini-flash-1.5",
98
+ "meta-llama/llama-3-70b-instruct",
99
+ "meta-llama/llama-3-8b-instruct",
100
+
101
  ]
102
  class FollowupQueryModel(BaseModel):
103
  query: str = Field(..., description="User's query for the followup agent")