Nioi commited on
Commit
1de5d30
·
1 Parent(s): 70fdc68

minor fixes on parameters and api endpoint

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -13,7 +13,7 @@ def get_agent() -> CodeAgent:
13
 
14
  model = OpenAIServerModel(
15
  model_id='codestral-latest',
16
- api_base="https://codestral.mistral.ai/v1/",
17
  api_key=api_key)
18
 
19
- return CodeAgent(tools=[search_tool, web_page_tool, add, subtract, multiply, divide, modulus, rounder, power, square_root, download_file_from_url, csv_reader, excel_reader, transcribe_audio, wiki_search, arvix_search], model=model, planning_interval=None, additional_authorized_imports=['random', 'time'], verbosity_level=1, max_steps=5, add_base_tools=True)
 
13
 
14
  model = OpenAIServerModel(
15
  model_id='codestral-latest',
16
+ api_base="https://codestral.mistral.ai/v1/chat/completions",
17
  api_key=api_key)
18
 
19
+ return CodeAgent(tools=[search_tool, web_page_tool, add, subtract, multiply, divide, modulus, rounder, power, square_root, download_file_from_url, csv_reader, excel_reader, transcribe_audio, wiki_search, arvix_search], model=model, additional_authorized_imports=['random', 'time'], add_base_tools=True)