init

api/doubao_client.py

@@ -0,0 +1,36 @@
+import os
+import logging
+from openai import OpenAI
+from typing import Optional, Dict, Any
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+class DoubaoClient():
+    def __init__(self, api_key: str = None):
+        self.api_key="f6150e6c-422a-4265-8b63-4d941b271220"
+        self.model="doubao-1.5-vision-pro-250328"
+        self.api_url="https://ark.cn-beijing.volces.com/api/v3"
+        self.client = OpenAI(
+            base_url=self.api_url,
+            api_key=self.api_key,
+        )
+        logging.info("DoubaoClient initialized.")
+
+    def chat_completion(self, messages: list, **kwargs) -> Optional[str]:
+        """Specific chat completion method for Doubao using OpenAI client."""
+        try:
+            response = self.client.chat.completions.create(
+                model=self.model,
+                messages=messages,
+                **kwargs
+            )
+            if response.choices and len(response.choices) > 0:
+                return response.choices[0].message.content
+            else:
+                logging.error("Failed to get content from Doubao API response.")
+                return None
+        except Exception as e:
+            logging.error(f"Doubao API call failed: {e}")
+            return None