UntilDot commited on
Commit
c5acba3
·
verified ·
1 Parent(s): 1e63fdd

Update llm/agents.py

Browse files
Files changed (1) hide show
  1. llm/agents.py +4 -0
llm/agents.py CHANGED
@@ -39,6 +39,10 @@ async def query_llm_agent(name: str, prompt: str, settings: dict) -> str:
39
  if not selected_model:
40
  return f"[{name}] No model selected."
41
 
 
 
 
 
42
  if selected_model not in ALLOWED_MODELS:
43
  return f"[{name}] Model '{selected_model}' is not supported."
44
 
 
39
  if not selected_model:
40
  return f"[{name}] No model selected."
41
 
42
+ # Auto-append ":free" if missing
43
+ if not selected_model.endswith(":free"):
44
+ selected_model += ":free"
45
+
46
  if selected_model not in ALLOWED_MODELS:
47
  return f"[{name}] Model '{selected_model}' is not supported."
48