fix: llm session tweaks & agent init refactor

This commit is contained in:
Liang Jiaqing
2026-03-22 09:55:24 +08:00
parent 122b6c431c
commit a9f0e9628f
2 changed files with 4 additions and 4 deletions

View File

@@ -234,7 +234,7 @@ class LLMSession:
streamed_any = True
yield delta; buffer += delta
if finish_reason: break
if '</tool_use>' in buffer[-30:]: break
#if '</tool_use>' in buffer[-30:]: break
return
except requests.HTTPError as e:
resp = getattr(e, "response", None)
@@ -331,7 +331,7 @@ class LLMSession:
class GeminiSession:
def __init__(self, cfg):
self.api_key = cfg.get('apikey') or google_api_key
self.api_key = cfg.get('apikey')
if not self.api_key: raise ValueError("google_api_key 未配置或为空,请在 mykey.py 中设置")
self.default_model = cfg.get('model', 'gemini-2.0-flash-001')
p = cfg.get('proxy', proxy)