update: upload fix

backend/app/services/llm/deepseek_vision.py

@@ -9,26 +9,39 @@ from app.services.llm.base import BaseLLMProvider
 from app.prompts.extract_transaction import get_extract_messages
 from app.services.llm.openai_vision import _parse_json_array
 
-
-# DeepSeek vision endpoint (OpenAI-compatible)
 DEEPSEEK_BASE = "https://api.deepseek.com"
 
 
 class DeepSeekVisionProvider(BaseLLMProvider):
-    async def extract_from_image(self, image_bytes: bytes) -> list[TransactionExtractItem]:
+    def _get_client(self) -> AsyncOpenAI:
         settings = get_settings()
         if not settings.deepseek_api_key:
             raise ValueError("DEEPSEEK_API_KEY is not set")
-        client = AsyncOpenAI(
-            api_key=settings.deepseek_api_key,
-            base_url=DEEPSEEK_BASE,
-        )
+        return AsyncOpenAI(api_key=settings.deepseek_api_key, base_url=DEEPSEEK_BASE)
+
+    def _get_model(self) -> str:
+        return self._model_override or get_settings().deepseek_model
+
+    async def extract_from_image(self, image_bytes: bytes) -> list[TransactionExtractItem]:
+        client = self._get_client()
         b64 = base64.standard_b64encode(image_bytes).decode("ascii")
         messages = get_extract_messages(b64)
         response = await client.chat.completions.create(
-            model=settings.deepseek_model,
+            model=self._get_model(),
             messages=messages,
             max_tokens=4096,
         )
         text = response.choices[0].message.content or "[]"
         return _parse_json_array(text)
+
+    async def chat(self, system: str, user: str) -> str:
+        client = self._get_client()
+        response = await client.chat.completions.create(
+            model=self._get_model(),
+            messages=[
+                {"role": "system", "content": system},
+                {"role": "user", "content": user},
+            ],
+            max_tokens=4096,
+        )
+        return response.choices[0].message.content or ""