This commit is contained in:
fredqian
2025-04-25 17:23:28 +08:00
parent d32d91a1f6
commit 24ac8a8362
36 changed files with 56 additions and 27 deletions

View File

@@ -4,7 +4,7 @@ import json
class Ollama():
def __init__(self, url="", model="", max_history_num=0, max_tokens=1024):
def __init__(self, url="", model="", max_history_num=0):
self._heads = {
"Accept": "text/event-stream",
# "Cache-Control": "no-cache",
@@ -19,8 +19,7 @@ class Ollama():
self._data = {
"stream": True,
"model": model,
"messages": self._messages,
"max_tokens": max_tokens
"messages": self._messages
}
def set_timeout(self, timeout):