diff --git a/src/models/chat_model.py b/src/models/chat_model.py index 8df94f7b..f6b8a8fc 100644 --- a/src/models/chat_model.py +++ b/src/models/chat_model.py @@ -67,15 +67,13 @@ class QianfanResponse: class Qianfan: - def __init__(self, model_name="ernie-lite-8k") -> None: + def __init__(self, model_name="ernie_speed") -> None: self.model_name = model_name access_key = os.getenv("QIANFAN_ACCESS_KEY") secret_key = os.getenv("QIANFAN_SECRET_KEY") self.client = qianfan.ChatCompletion(ak=access_key, sk=secret_key) def predict(self, message, stream=False): - - logger.debug(message) if isinstance(message, str): messages=[{"role": "user", "content": message}] else: diff --git a/src/views/common_view.py b/src/views/common_view.py index 8d0e989d..8e0b70dd 100644 --- a/src/views/common_view.py +++ b/src/views/common_view.py @@ -58,8 +58,8 @@ def chat(): def call(): request_data = json.loads(request.data) query = request_data['query'] - logger.debug(f"Web query: {query}") response = model.predict(query) + logger.debug(f"Call query: {query} Response: {response.content}") return jsonify({ "response": response.content,