github/autoMate
1
0
Fork 0
mirror of https://github.com/yuruotong1/autoMate.git synced 2025-12-26 05:16:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #58 from yuruotong1/dev

Browse Source 
更新 token
This commit is contained in:
Dongle 2025-03-07 09:15:48 +08:00 committed by GitHub
commit f91e905069
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gradio_ui/agent/vlm_agent.py
View File

@ -85,13 +85,12 @@ class VLMAgent:
provider_base_url=self.base_url,
temperature=0,
)
self.total_token_usage += token_usage
latency_vlm = time.time() - start
self.output_callback(f"LLM: {latency_vlm:.2f}s, OmniParser: {latency_omniparser:.2f}s", sender="bot")
print(f"llm_response: {vlm_response}")
if self.print_usage:
print(f"Total token so far: {self.total_token_usage}. Total cost so far: $USD{self.total_cost:.5f}")
print(f"Total token so far: {token_usage}. Total cost so far: $USD{self.total_cost:.5f}")
vlm_response_json = extract_data(vlm_response, "json")
vlm_response_json = json.loads(vlm_response_json)