Spaces:
Runtime error
Runtime error
starsnatched
commited on
Commit
·
e98e0c1
1
Parent(s):
db3e3e2
Remove unused EMBEDDING_MODEL_NAME from ChatSession initialization
Browse files- src/chat.py +0 -2
src/chat.py
CHANGED
|
@@ -11,7 +11,6 @@ from ollama import AsyncClient, ChatResponse, Message
|
|
| 11 |
from .config import (
|
| 12 |
MAX_TOOL_CALL_DEPTH,
|
| 13 |
MODEL_NAME,
|
| 14 |
-
EMBEDDING_MODEL_NAME,
|
| 15 |
NUM_CTX,
|
| 16 |
OLLAMA_HOST,
|
| 17 |
SYSTEM_PROMPT,
|
|
@@ -40,7 +39,6 @@ class ChatSession:
|
|
| 40 |
session: str = "default",
|
| 41 |
host: str = OLLAMA_HOST,
|
| 42 |
model: str = MODEL_NAME,
|
| 43 |
-
embedding_model: str = EMBEDDING_MODEL_NAME,
|
| 44 |
) -> None:
|
| 45 |
init_db()
|
| 46 |
self._client = AsyncClient(host=host)
|
|
|
|
| 11 |
from .config import (
|
| 12 |
MAX_TOOL_CALL_DEPTH,
|
| 13 |
MODEL_NAME,
|
|
|
|
| 14 |
NUM_CTX,
|
| 15 |
OLLAMA_HOST,
|
| 16 |
SYSTEM_PROMPT,
|
|
|
|
| 39 |
session: str = "default",
|
| 40 |
host: str = OLLAMA_HOST,
|
| 41 |
model: str = MODEL_NAME,
|
|
|
|
| 42 |
) -> None:
|
| 43 |
init_db()
|
| 44 |
self._client = AsyncClient(host=host)
|