tech-envision commited on
Commit
e32caeb
·
unverified ·
2 Parent(s): 79d68f3 4c7374d

Merge pull request #8 from tech-envision/codex/add-num_ctx-option-to-config.py

Browse files
Files changed (2) hide show
  1. src/chat.py +2 -1
  2. src/config.py +1 -0
src/chat.py CHANGED
@@ -5,7 +5,7 @@ import json
5
 
6
  from ollama import AsyncClient, ChatResponse, Message
7
 
8
- from .config import MAX_TOOL_CALL_DEPTH, MODEL_NAME, OLLAMA_HOST
9
  from .db import Conversation, Message as DBMessage, User, _db, init_db
10
  from .log import get_logger
11
  from .schema import Msg
@@ -78,6 +78,7 @@ class ChatSession:
78
  messages=messages,
79
  think=think,
80
  tools=[add_two_numbers, execute_python],
 
81
  )
82
 
83
  async def _handle_tool_calls(
 
5
 
6
  from ollama import AsyncClient, ChatResponse, Message
7
 
8
+ from .config import MAX_TOOL_CALL_DEPTH, MODEL_NAME, NUM_CTX, OLLAMA_HOST
9
  from .db import Conversation, Message as DBMessage, User, _db, init_db
10
  from .log import get_logger
11
  from .schema import Msg
 
78
  messages=messages,
79
  think=think,
80
  tools=[add_two_numbers, execute_python],
81
+ options={"num_ctx": NUM_CTX},
82
  )
83
 
84
  async def _handle_tool_calls(
src/config.py CHANGED
@@ -6,3 +6,4 @@ from typing import Final
6
  MODEL_NAME: Final[str] = os.getenv("OLLAMA_MODEL", "qwen3")
7
  OLLAMA_HOST: Final[str] = os.getenv("OLLAMA_HOST", "http://localhost:11434")
8
  MAX_TOOL_CALL_DEPTH: Final[int] = 5
 
 
6
  MODEL_NAME: Final[str] = os.getenv("OLLAMA_MODEL", "qwen3")
7
  OLLAMA_HOST: Final[str] = os.getenv("OLLAMA_HOST", "http://localhost:11434")
8
  MAX_TOOL_CALL_DEPTH: Final[int] = 5
9
+ NUM_CTX: Final[int] = int(os.getenv("OLLAMA_NUM_CTX", "4096"))