JPM34 commited on
Commit
7e0e19c
·
1 Parent(s): 4bb4d29

Updated to llama4

Browse files
Files changed (2) hide show
  1. agent.py +2 -2
  2. models.py +2 -0
agent.py CHANGED
@@ -62,8 +62,8 @@ class BasicAgent:
62
  def __init__(
63
  self,
64
  verbose=False,
65
- name_model_provider="mistral",
66
- name_model=None,
67
  ):
68
  self.verbose = verbose
69
  # tools = [
 
62
  def __init__(
63
  self,
64
  verbose=False,
65
+ name_model_provider="openrouter",
66
+ name_model="llama4",
67
  ):
68
  self.verbose = verbose
69
  # tools = [
models.py CHANGED
@@ -16,6 +16,8 @@ def select_model(provider, name_model=None):
16
  reference_model = "google/gemini-2.5-pro-exp-03-25"
17
  case "deepseek":
18
  reference_model = "deepseek/deepseek-chat-v3-0324:free"
 
 
19
  case _:
20
  print("Invalid command.")
21
  api_key = os.getenv("OPENROUTER_API_KEY")
 
16
  reference_model = "google/gemini-2.5-pro-exp-03-25"
17
  case "deepseek":
18
  reference_model = "deepseek/deepseek-chat-v3-0324:free"
19
+ case "llama4":
20
+ reference_model = "meta-llama/llama-4-scout:free"
21
  case _:
22
  print("Invalid command.")
23
  api_key = os.getenv("OPENROUTER_API_KEY")