fix intention

models/registry.py

@@ -0,0 +1,13 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/12/26 17:52
+@Author  : cao
+@File    : registry.py
+@Desc    : 
+"""
+from prometheus_client import Counter
+REQUEST_COUNT = Counter('others', 'intention for others')
+
+
+

models/web.py

@@ -3,7 +3,7 @@
 """
 @Time    : 2024/9/10 18:33
 @Author  : cao
-@File    : web.py.py
+@File    : web2.py.py
 @Desc    : 
 """
 import json
@@ -19,6 +19,9 @@ import logging
 from starlette.responses import JSONResponse
 import dirty
 from prometheus_fastapi_instrumentator import Instrumentator, metrics
+# from prometheus_client import Counter as myCounter
+# from prometheus_client import REGISTRY,gc_collector, platform_collector, process_collector
+
 
 logger = logging.getLogger('my_logger')
 logger.setLevel(logging.INFO)
@@ -26,7 +29,6 @@ console_handler = logging.StreamHandler()
 formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 logger.addHandler(console_handler)
 
-
 model = utils.load_model()
 index = utils.index
 cate = utils.cate
@@ -37,6 +39,7 @@ class Sentence(BaseModel):
 
 app = FastAPI()
 Instrumentator().instrument(app).expose(app)
+from registry import REQUEST_COUNT
 
 @app.post('/intention')
 def intention(sentence:Sentence):
@@ -87,6 +90,7 @@ def intention(sentence:Sentence):
         res['query'] = query
         if out[0]['prob'] < 0.7:
             res['intent'] = 'others'
+            REQUEST_COUNT.inc()
             res['status'] = 4
         else:
             res['intent'] = out[0]['cate']
@@ -96,7 +100,4 @@ def intention(sentence:Sentence):
 
 
 if __name__ == '__main__':
-    uvicorn.run("web:app", host='0.0.0.0', port=50072, workers=4)
-
-
-
+    uvicorn.run("web:app", host='0.0.0.0', port=50072, workers=4)