Spaces:
Runtime error
Runtime error
tech-envision
commited on
Commit
·
4c7374d
1
Parent(s):
79d68f3
Add configurable num_ctx option
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"))
|