shekkari21 commited on
Commit
d2aa9b5
·
1 Parent(s): 8d5b1f0

made changes

Browse files
Files changed (1) hide show
  1. config.py +7 -7
config.py CHANGED
@@ -62,14 +62,14 @@ def get_llm() -> LLM:
62
  # Free tier available: https://openrouter.ai
63
  # Note: For Hugging Face Spaces, set OPENROUTER_API_KEY as a secret
64
  if not OPENROUTER_API_KEY:
65
- # Try to use without key (some free models might work)
66
- # But it's better to set the key for reliability
67
- print("Warning: OPENROUTER_API_KEY not set. Some models may not work.")
68
- print("Get a free key at https://openrouter.ai")
 
69
  return LLM(
70
- model=OPENROUTER_MODEL,
71
- base_url="https://openrouter.ai/api/v1",
72
- api_key=OPENROUTER_API_KEY if OPENROUTER_API_KEY else "free"
73
  )
74
 
75
  else:
 
62
  # Free tier available: https://openrouter.ai
63
  # Note: For Hugging Face Spaces, set OPENROUTER_API_KEY as a secret
64
  if not OPENROUTER_API_KEY:
65
+ raise ValueError(
66
+ "OPENROUTER_API_KEY environment variable is not set. "
67
+ "Get a free key at https://openrouter.ai"
68
+ )
69
+ # CrewAI uses LiteLLM internally, so we need to format as openrouter/model-name
70
  return LLM(
71
+ model=f"openrouter/{OPENROUTER_MODEL}",
72
+ api_key=OPENROUTER_API_KEY
 
73
  )
74
 
75
  else: