余尚辉 2 months ago
parent
commit
0c835d9b1a
2 changed files with 9 additions and 9 deletions
  1. 2 2
      ai-slibra-assistant/api/AiBot.py
  2. 7 7
      ai-slibra-assistant/util/registry.py

+ 2 - 2
ai-slibra-assistant/api/AiBot.py

@@ -25,7 +25,7 @@ from entity import Error
 import traceback
 from util import insert_log, timetic
 from prometheus_fastapi_instrumentator import Instrumentator
-from prometheus_client import Counter,Histogram
+from prometheus_client import Histogram
 import time
 
 
@@ -89,8 +89,8 @@ def botservice(reqbot:reqRobot):
         #del_cache(uid, bid, 'status', session_id)
         return ChatResponse(code=Error.error.value, message="internal error").to_json_string()
     finally:
-        print('ceshi')
         latency = (time.time() - start_time)
+        print(f"Latency for {bid}: {latency}")  # 日志记录
         AI_BOT_REQUEST_LATENCY.labels(bid=bid).observe(latency)
 
 

+ 7 - 7
ai-slibra-assistant/util/registry.py

@@ -1,7 +1,7 @@
-from api.AiBot import app
-from prometheus_fastapi_instrumentator import Instrumentator
-from prometheus_client import Counter,Histogram
-
-Instrumentator().instrument(app).expose(app)
-
-AI_BOT_REQUEST_LATENCY = Histogram('ai_bot_request_latency',  '文本机器人接口耗时',['bid'])
+# from api.AiBot import app
+# from prometheus_fastapi_instrumentator import Instrumentator
+# from prometheus_client import Counter,Histogram
+#
+# Instrumentator().instrument(app).expose(app)
+#
+# AI_BOT_REQUEST_LATENCY = Histogram('ai_bot_request_latency',  '文本机器人接口耗时',['bid'])