petrov826 commited on
Commit
61a953b
·
verified ·
1 Parent(s): 14bf7f9

stop hard-coding model_id

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,8 +28,8 @@ rate_limiter = RateLimiter()
28
  class BasicAgent:
29
  def __init__(self):
30
  model = LiteLLMModel(
31
- model_id="gemini/gemini-2.5-flash",
32
- api_key=os.getenv("GOOGLE_API_KEY")
33
  )
34
  self.agent = CodeAgent(
35
  tools=[
 
28
  class BasicAgent:
29
  def __init__(self):
30
  model = LiteLLMModel(
31
+ model_id=os.getenv("MODEL_ID"),
32
+ api_key=os.getenv("API_KEY")
33
  )
34
  self.agent = CodeAgent(
35
  tools=[