Spaces:
Runtime error
Runtime error
Merge pull request #8 from tech-envision/codex/add-num_ctx-option-to-config.py
Browse files- src/chat.py +2 -1
- 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"))
|