From a9f0e9628f2642608fb13c839c37d1a3e03cc366 Mon Sep 17 00:00:00 2001 From: Liang Jiaqing Date: Sun, 22 Mar 2026 09:55:24 +0800 Subject: [PATCH] fix: llm session tweaks & agent init refactor --- agentmain.py | 4 ++-- llmcore.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/agentmain.py b/agentmain.py index ff6e3d6..d990bd7 100644 --- a/agentmain.py +++ b/agentmain.py @@ -55,7 +55,7 @@ class GeneraticAgent: self.lock = threading.Lock() self.history = [] self.task_queue = queue.Queue() - self.is_running, self.stop_sig = False, False + self.is_running = False; self.stop_sig = False self.llm_no = 0; self.inc_out = False self.handler = None; self.verbose = True self.llmclient = self.llmclients[self.llm_no] @@ -141,7 +141,7 @@ if __name__ == '__main__': args = parser.parse_args() agent = GeneraticAgent() - agent.llm_no = args.llm_no + agent.next_llm(args.llm_no) agent.verbose = False threading.Thread(target=agent.run, daemon=True).start() diff --git a/llmcore.py b/llmcore.py index 1d85a8e..f46f8d6 100644 --- a/llmcore.py +++ b/llmcore.py @@ -234,7 +234,7 @@ class LLMSession: streamed_any = True yield delta; buffer += delta if finish_reason: break - if '' in buffer[-30:]: break + #if '' 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)