starsnatched commited on
Commit
c56348f
·
1 Parent(s): ec09df9

Update model name and adjust context size in configuration

Browse files
Files changed (2) hide show
  1. run.py +4 -2
  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("note.pdf")
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", "32768"))
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"