minor fixes on parameters and api endpoint
Browse files
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,
|
|
|
|
| 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)
|