22Nikk0 commited on
Commit
3940dab
·
verified ·
1 Parent(s): 8e868e6

Update agentic.py

Browse files
Files changed (1) hide show
  1. agentic.py +3 -2
agentic.py CHANGED
@@ -50,7 +50,7 @@ def get_assistant_model():
50
 
51
  if llm_provider == "mistral":
52
  assistant_model = ChatMistralAI(
53
- model="mistral-large-2411",#"mistral-small-latest",#"ministral-8b-latest",#
54
  temperature=0,
55
  max_retries=2,
56
  api_key=os.getenv("MISTRAL_API_KEY")
@@ -404,7 +404,8 @@ if __name__ == "__main__":
404
 
405
  file_name = ""
406
 
407
- question = "In the video https://www.youtube.com/watch?v=L1vXCYZAYYM, what is the highest number of bird species to be on camera simultaneously?"
 
408
  # file_name = question_json.get("file_name", "")
409
 
410
  print(f"QUESTION : {question}")
 
50
 
51
  if llm_provider == "mistral":
52
  assistant_model = ChatMistralAI(
53
+ model="mistral-small-latest",#"ministral-8b-latest",#"mistral-large-2411",#
54
  temperature=0,
55
  max_retries=2,
56
  api_key=os.getenv("MISTRAL_API_KEY")
 
404
 
405
  file_name = ""
406
 
407
+ # question = "In the video https://www.youtube.com/watch?v=L1vXCYZAYYM, what is the highest number of bird species to be on camera simultaneously?"
408
+ question = "Who are the pitchers with the number before and after Taishō Tamai's number as of July 2023? Give them to me in the form Pitcher Before, Pitcher After, use their last names only, in Roman characters."
409
  # file_name = question_json.get("file_name", "")
410
 
411
  print(f"QUESTION : {question}")