Spaces:
Runtime error
Runtime error
russkiyximik commited on
Commit ·
ee375ec
1
Parent(s): 20665e6
changed litellm config
Browse files
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=
|
| 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 |
)
|