Browse Source

feat: update

jerry 2 months ago
parent
commit
e5a1eab6e0
2 changed files with 7 additions and 4 deletions
  1. 1 1
      .env
  2. 6 3
      app/services/llm_service.py

+ 1 - 1
.env

@@ -1,6 +1,6 @@
 ENV=DEV
 ENV=DEV
 DATABASE_URL=mysql+asyncmy://root:GqLLL7Bofj0WaaOpp.0@visafly.top:3306/book_user_info?charset=utf8mb4
 DATABASE_URL=mysql+asyncmy://root:GqLLL7Bofj0WaaOpp.0@visafly.top:3306/book_user_info?charset=utf8mb4
 REDIS_URL=redis://:STEs2x6ML0U1HlpE9SojM6YU7QPhqzY8@45.137.220.138:6379/0
 REDIS_URL=redis://:STEs2x6ML0U1HlpE9SojM6YU7QPhqzY8@45.137.220.138:6379/0
-OPENAI_API_KEY=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
+OPENAI_API_KEY=sk-893e895724c6403d81374e515ffaf427
 STRIPE_API_KEY=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
 STRIPE_API_KEY=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
 STRIPE_WEBHOOK_SECRET=whsec_MRfGMJZVv1v8xnwnGl9ai8yXqpLSeQVL
 STRIPE_WEBHOOK_SECRET=whsec_MRfGMJZVv1v8xnwnGl9ai8yXqpLSeQVL

+ 6 - 3
app/services/llm_service.py

@@ -5,14 +5,16 @@ import os
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy import select
 from sqlalchemy import select
 from app.core.config import settings
 from app.core.config import settings
+from app.core.biz_exception import NotFoundError
 from app.schemas.llm import ParseUserInputsPayload, ParseUserInputsOut
 from app.schemas.llm import ParseUserInputsPayload, ParseUserInputsOut
 from app.models.schema import VasSchema
 from app.models.schema import VasSchema
 
 
 # --- 配置区 ---
 # --- 配置区 ---
 # 请换成你新生成的 API Key
 # 请换成你新生成的 API Key
 API_KEY = settings.openai_api_key
 API_KEY = settings.openai_api_key
-API_URL = "https://api.openai.com/v1/chat/completions"
 
 
+# API_URL = "https://api.openai.com/v1/chat/completions"
+API_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions"
 
 
 class LlmService:
 class LlmService:
 
 
@@ -49,7 +51,8 @@ class LlmService:
         """
         """
 
 
         payload = {
         payload = {
-            "model": "gpt-4o",  # 或 gpt-3.5-turbo
+            # "model": "gpt-4o",  # 或 gpt-3.5-turbo
+            "model": "qwen-plus",
             "messages": [
             "messages": [
                 {"role": "system", "content": system_instruction},
                 {"role": "system", "content": system_instruction},
                 {"role": "user", "content": user_prompt}
                 {"role": "user", "content": user_prompt}
@@ -91,5 +94,5 @@ if __name__ == "__main__":
     user_input = "我是张三,打算花2000块钱买个耳机和键盘。"
     user_input = "我是张三,打算花2000块钱买个耳机和键盘。"
     
     
     print("正在解析...")
     print("正在解析...")
-    result = parse_data_api(user_input, my_schema)
+    result = LlmService.parse_data_async(user_input, my_schema)
     print(json.dumps(result, ensure_ascii=False, indent=2))
     print(json.dumps(result, ensure_ascii=False, indent=2))