优化 Ollama的初始化

This commit is contained in:
dahanzimin
2025-03-09 20:29:35 +08:00
parent 445be6f7f8
commit 4ec90645ce

View File

@@ -4,7 +4,7 @@ import json
class Ollama():
def __init__(self, url="", model=""):
def __init__(self, url="", model="", max_history_num=0):
self._heads = {
"Accept": "text/event-stream",
# "Cache-Control": "no-cache",
@@ -13,7 +13,7 @@ class Ollama():
}
self._url = url
self._max_retries = 1
self._max_history_num = 0
self._max_history_num = max_history_num
self._timeout = 10000
self._messages = []
self._data = {