Spaces:
Runtime error
Runtime error
starsnatched
commited on
Commit
·
c56348f
1
Parent(s):
ec09df9
Update model name and adjust context size in configuration
Browse files- run.py +4 -2
- src/config.py +2 -2
run.py
CHANGED
@@ -8,8 +8,10 @@ from src.vm import VMRegistry
|
|
8 |
|
9 |
async def _main() -> None:
|
10 |
async with TeamChatSession(user="demo_user", session="demo_session") as chat:
|
11 |
-
# doc_path = chat.upload_document("
|
12 |
-
async for resp in chat.chat_stream("ask how junior agent is doing"):
|
|
|
|
|
13 |
print("\n>>>", resp)
|
14 |
|
15 |
|
|
|
8 |
|
9 |
async def _main() -> None:
|
10 |
async with TeamChatSession(user="demo_user", session="demo_session") as chat:
|
11 |
+
# doc_path = chat.upload_document("requirements.txt")
|
12 |
+
# async for resp in chat.chat_stream("ask how junior agent is doing"):
|
13 |
+
# async for resp in chat.chat_stream("what is in the requirements.txt file in /data?"):
|
14 |
+
async for resp in chat.chat_stream("add transformers package to requirements.txt."):
|
15 |
print("\n>>>", resp)
|
16 |
|
17 |
|
src/config.py
CHANGED
@@ -6,10 +6,10 @@ from typing import Final
|
|
6 |
from dotenv import load_dotenv
|
7 |
load_dotenv()
|
8 |
|
9 |
-
MODEL_NAME: Final[str] = os.getenv("OLLAMA_MODEL", "qwen3")
|
10 |
OLLAMA_HOST: Final[str] = os.getenv("OLLAMA_HOST", "http://localhost:11434")
|
11 |
MAX_TOOL_CALL_DEPTH: Final[int] = 15
|
12 |
-
NUM_CTX: Final[int] = int(os.getenv("OLLAMA_NUM_CTX", "
|
13 |
UPLOAD_DIR: Final[str] = os.getenv("UPLOAD_DIR", str(Path.cwd() / "uploads"))
|
14 |
VM_IMAGE: Final[str] = os.getenv("VM_IMAGE", "python:3.11")
|
15 |
PERSIST_VMS: Final[bool] = os.getenv("PERSIST_VMS", "1") == "1"
|
|
|
6 |
from dotenv import load_dotenv
|
7 |
load_dotenv()
|
8 |
|
9 |
+
MODEL_NAME: Final[str] = os.getenv("OLLAMA_MODEL", "qwen3:1.7b")
|
10 |
OLLAMA_HOST: Final[str] = os.getenv("OLLAMA_HOST", "http://localhost:11434")
|
11 |
MAX_TOOL_CALL_DEPTH: Final[int] = 15
|
12 |
+
NUM_CTX: Final[int] = int(os.getenv("OLLAMA_NUM_CTX", "16384"))
|
13 |
UPLOAD_DIR: Final[str] = os.getenv("UPLOAD_DIR", str(Path.cwd() / "uploads"))
|
14 |
VM_IMAGE: Final[str] = os.getenv("VM_IMAGE", "python:3.11")
|
15 |
PERSIST_VMS: Final[bool] = os.getenv("PERSIST_VMS", "1") == "1"
|