ys/GenericAgent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: llm session tweaks & agent init refactor

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
agentmain.py
View File

@@ -55,7 +55,7 @@ class GeneraticAgent:
self.lock = threading.Lock() self.lock = threading.Lock()
self.history = [] self.history = []
self.task_queue = queue.Queue() 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.llm_no = 0; self.inc_out = False
self.handler = None; self.verbose = True self.handler = None; self.verbose = True
self.llmclient = self.llmclients[self.llm_no] self.llmclient = self.llmclients[self.llm_no]
@@ -141,7 +141,7 @@ if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
agent = GeneraticAgent() agent = GeneraticAgent()
agent.llm_no = args.llm_no agent.next_llm(args.llm_no)
agent.verbose = False agent.verbose = False
threading.Thread(target=agent.run, daemon=True).start() threading.Thread(target=agent.run, daemon=True).start()

4
llmcore.py
View File

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