langchain message to openai

routers/openai_v1.py

@@ -1,7 +1,7 @@
 from fastapi import APIRouter, Depends
-import os
-from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import ChatPromptTemplate
+from datetime import datetime
+# import uuid
 
 from global_state import get
 from db.tbs_db import TbsDb
@@ -45,15 +45,16 @@ async def chat_completions(chat_model:ChatModel, current_user: UserModel = Depen
     
     # 生成prompt模板
     lc_messages = [(message.role, message.content) for message in chat_model.messages]
-    parser = StrOutputParser()
     prompt_template = ChatPromptTemplate.from_messages(lc_messages)
- 
-    chain = prompt_template | llm | parser
+    chain = prompt_template | llm
     try:
-        result = chain.invoke({})
+        result = chain.invoke({}) # AIMessage 类对象
     except Exception as e:
         return {'error': str(e)}
-    return result
+    
+    # 转换为OpenAI格式
+    converted_data = convert_to_openai_format(result)
+    return converted_data
 
 # 从数据库获取默认模型
 async def get_default_model():
@@ -76,9 +77,6 @@ ORDER BY ak.last_call_at
 limit 1
 """
     response = TbsDb(db_module_filename, "Cloudflare").get_item(query)
-    # return response
-    # results = Db(os.getenv("DB_PATH")).list_query(sql)
-    # return results
     try:
         result = response['result'][0]['results'][0]
         api_key = result['api_key']
@@ -89,11 +87,26 @@ limit 1
     TbsDb(db_module_filename, "Cloudflare").execute_query(query)
     return result
 
-async def get_user_id(api_key):
-    sql = f"select id from users where api_key='{api_key}'"
-    results = Db(os.getenv("DB_PATH")).list_query(sql)
-    try:
-        result = results[0]['id']
-    except:
-        result = 0
-    return result
+def convert_to_openai_format(original_json):
+    # 创建新的JSON对象
+    new_json = {
+        "id": "chatcmpl-123",  # 这里可以生成一个唯一的ID，或者使用传入的id
+        "object": "chat.completion",
+        "created": int(datetime.now().timestamp()),  # 当前时间戳
+        "choices": [
+            {
+                "index": 0,
+                "message": {
+                    "role": "assistant",
+                    "content": original_json.content  # 使用原始内容
+                },
+                "finish_reason": "stop"
+            }
+        ],
+        "usage": {
+            "prompt_tokens": original_json.usage_metadata.get("input_tokens",0),
+            "completion_tokens": original_json.usage_metadata.get("output_tokens", 0),
+            "total_tokens": original_json.usage_metadata.get("total_tokens", 0)
+        }
+    }
+    return new_json