JarvisChan630 commited on
Commit
fc95199
1 Parent(s): 6e732f0
Files changed (3) hide show
  1. agents/meta_agent.py +1 -1
  2. chat.py +1 -1
  3. models/llms.py +1 -1
agents/meta_agent.py CHANGED
@@ -399,7 +399,7 @@ if __name__ == "__main__":
399
 
400
  # For Claude
401
  agent_kwargs = {
402
- "model": "claude-3-5-sonnet-20240620",
403
  "server": "claude",
404
  "temperature": 0.5
405
  }
 
399
 
400
  # For Claude
401
  agent_kwargs = {
402
+ "model": "claude-3-haiku-20240307",
403
  "server": "claude",
404
  "temperature": 0.5
405
  }
chat.py CHANGED
@@ -58,7 +58,7 @@ def get_agent_kwargs(server: str = "claude", location: str = None, hybrid: bool
58
 
59
  elif server == "claude":
60
  agent_kwargs = {
61
- "model": "claude-3-5-sonnet-20240620",
62
  "server": "claude",
63
  "temperature": 0,
64
  }
 
58
 
59
  elif server == "claude":
60
  agent_kwargs = {
61
+ "model": "claude-3-haiku-20240307",
62
  "server": "claude",
63
  "temperature": 0,
64
  }
models/llms.py CHANGED
@@ -103,7 +103,7 @@ class ClaudeModel(BaseModel):
103
  'x-api-key': self.api_key,
104
  'anthropic-version': '2023-06-01'
105
  }
106
- self.model_endpoint = "https://api.anthropic.com/v1/messages"
107
 
108
  def invoke(self, messages: List[Dict[str, str]]) -> str:
109
  # time.sleep(5)
 
103
  'x-api-key': self.api_key,
104
  'anthropic-version': '2023-06-01'
105
  }
106
+ self.model_endpoint = "https://api.302.ai/v1/chat/completions"
107
 
108
  def invoke(self, messages: List[Dict[str, str]]) -> str:
109
  # time.sleep(5)