russkiyximik commited on
Commit
ee375ec
·
1 Parent(s): 20665e6

changed litellm config

Browse files
Files changed (1) hide show
  1. mycoder/src/inner/crew.py +3 -2
mycoder/src/inner/crew.py CHANGED
@@ -1,6 +1,7 @@
1
  from crewai import Agent, Crew, Process, Task, LLM
2
  from crewai.project import CrewBase, agent, crew, task, llm
3
 
 
4
 
5
 
6
  @CrewBase
@@ -15,7 +16,7 @@ class Coder():
15
 
16
  @llm
17
  def ollama_llm(self):
18
- return LLM(model="ollama/codellama:7b", base_url="http://localhost:7860")
19
 
20
  @agent
21
  def coder(self) -> Agent:
@@ -27,7 +28,7 @@ class Coder():
27
  verbose=True,
28
  #allow_code_execution=True,
29
  #code_execution_mode="unsafe", # Currently doesn't use Docker for safety
30
- max_execution_time=1800,
31
  max_retry_limit=5 ,
32
  tools=[ExtractSchema(), code_interpreter],
33
  )
 
1
  from crewai import Agent, Crew, Process, Task, LLM
2
  from crewai.project import CrewBase, agent, crew, task, llm
3
 
4
+ MAX_TIMEOUT_SEC = 1800 # 30 mins, adjust as needed
5
 
6
 
7
  @CrewBase
 
16
 
17
  @llm
18
  def ollama_llm(self):
19
+ return LLM(model="ollama/codellama:7b", base_url="http://localhost:7860", timeout=MAX_TIMEOUT_SEC)
20
 
21
  @agent
22
  def coder(self) -> Agent:
 
28
  verbose=True,
29
  #allow_code_execution=True,
30
  #code_execution_mode="unsafe", # Currently doesn't use Docker for safety
31
+ max_execution_time=MAX_TIMEOUT_SEC,
32
  max_retry_limit=5 ,
33
  tools=[ExtractSchema(), code_interpreter],
34
  )