22Nikk0 commited on
Commit
e33b9dd
·
verified ·
1 Parent(s): 15a14e1

Update agentic.py

Browse files
Files changed (1) hide show
  1. agentic.py +2 -2
agentic.py CHANGED
@@ -62,7 +62,7 @@ def get_general_model():
62
 
63
  if llm_provider == "mistral":
64
  general_model = ChatMistralAI(
65
- model="ministral-8b-latest",#"mistral-small-latest",#"mistral-large-2411",#
66
  temperature=0,
67
  max_retries=2,
68
  api_key=os.getenv("MISTRAL_API_KEY")
@@ -257,7 +257,7 @@ def download_input_file(task_id: str) -> str:
257
 
258
  ######## LLM associations ########
259
 
260
- general_model = get_general_model()
261
  vision_model = get_vision_model()
262
  video_handler_model = get_video_handler_model()
263
  audio_handler_model = get_audio_handler_model()
 
62
 
63
  if llm_provider == "mistral":
64
  general_model = ChatMistralAI(
65
+ model="mistral-large-2411",#"ministral-8b-latest",#"mistral-small-latest",#
66
  temperature=0,
67
  max_retries=2,
68
  api_key=os.getenv("MISTRAL_API_KEY")
 
257
 
258
  ######## LLM associations ########
259
 
260
+ general_model = get_vision_model()
261
  vision_model = get_vision_model()
262
  video_handler_model = get_video_handler_model()
263
  audio_handler_model = get_audio_handler_model()