update: upload fix

backend/app/services/llm/openai_vision.py

@@ -12,6 +12,9 @@ from app.prompts.extract_transaction import get_extract_messages
 
 
 class OpenAIVisionProvider(BaseLLMProvider):
+    def _get_model(self) -> str:
+        return self._model_override or get_settings().openai_model
+
     async def extract_from_image(self, image_bytes: bytes) -> list[TransactionExtractItem]:
         settings = get_settings()
         if not settings.openai_api_key:
@@ -20,18 +23,32 @@ class OpenAIVisionProvider(BaseLLMProvider):
         b64 = base64.standard_b64encode(image_bytes).decode("ascii")
         messages = get_extract_messages(b64)
         response = await client.chat.completions.create(
-            model=settings.openai_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:
+        settings = get_settings()
+        if not settings.openai_api_key:
+            raise ValueError("OPENAI_API_KEY is not set")
+        client = AsyncOpenAI(api_key=settings.openai_api_key)
+        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 ""
+
 
 def _parse_json_array(text: str) -> list[TransactionExtractItem]:
     """Parse LLM response into list of TransactionExtractItem. Tolerates markdown and extra text."""
     text = text.strip()
-    # Remove optional markdown code block
     if text.startswith("```"):
         text = re.sub(r"^```(?:json)?\s*", "", text)
         text = re.sub(r"\s*```\s*$", "", text)
@@ -46,7 +63,6 @@ def _parse_json_array(text: str) -> list[TransactionExtractItem]:
         if not isinstance(item, dict):
             continue
         try:
-            # Normalize transaction_time: allow string -> datetime
             if isinstance(item.get("transaction_time"), str) and item["transaction_time"]:
                 from dateutil import parser as date_parser
                 item["transaction_time"] = date_parser.isoparse(item["transaction_time"])