余尚辉 2 months ago
parent
commit
d3e08b4612
2 changed files with 7 additions and 10 deletions
  1. 2 3
      ai-slibra-assistant/api/AiBot.py
  2. 5 7
      ai-slibra-assistant/util/registry.py

+ 2 - 3
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 Histogram
+
 import time
 
 
@@ -36,8 +36,7 @@ app = FastAPI()
 
 # 服务监控,qps,latent, cpu利用率
 Instrumentator().instrument(app).expose(app)
-# from util.registry import AI_BOT_REQUEST_LATENCY
-AI_BOT_REQUEST_LATENCY = Histogram('ai_bot_request_latency',  '文本机器人接口耗时',['bid'])
+from util.registry import AI_BOT_REQUEST_LATENCY
 
 
 @app.post("/botservice")

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

@@ -1,7 +1,5 @@
-# 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 prometheus_client import Counter,Histogram
+
+
+AI_BOT_REQUEST_LATENCY = Histogram('ai_bot_request_latency',  '文本机器人接口耗时',['bid'])