|
@@ -4,6 +4,8 @@
|
|
|
|
|
|
import json
|
|
import json
|
|
from typing import Dict, Any, Optional
|
|
from typing import Dict, Any, Optional
|
|
|
|
+
|
|
|
|
+import chat_service
|
|
from prompt_templates import USER_PROFILE_EXTRACT_PROMPT
|
|
from prompt_templates import USER_PROFILE_EXTRACT_PROMPT
|
|
from openai import OpenAI
|
|
from openai import OpenAI
|
|
import logging
|
|
import logging
|
|
@@ -14,10 +16,10 @@ import global_flags
|
|
class UserProfileExtractor:
|
|
class UserProfileExtractor:
|
|
def __init__(self):
|
|
def __init__(self):
|
|
self.llm_client = OpenAI(
|
|
self.llm_client = OpenAI(
|
|
- api_key='5e275c38-44fd-415f-abcf-4b59f6377f72',
|
|
|
|
- base_url="https://ark.cn-beijing.volces.com/api/v3"
|
|
|
|
|
|
+ api_key=chat_service.VOLCENGINE_API_TOKEN,
|
|
|
|
+ base_url=chat_service.VOLCENGINE_BASE_URL
|
|
)
|
|
)
|
|
- self.model_name = 'ep-20250307150409-4blz9'
|
|
|
|
|
|
+ self.model_name = chat_service.VOLCENGINE_MODEL_DOUBAO_PRO_1_5
|
|
|
|
|
|
def get_extraction_function(self) -> Dict:
|
|
def get_extraction_function(self) -> Dict:
|
|
"""
|
|
"""
|