fix: (models, sync) add error handling and logging for OpenAI respons…

…es and data filtering
yufeikang · Jul 5, 2024 · 257c229 · 257c229
1 parent 2a163e7
commit 257c229
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/app/models.py b/app/models.py
@@ -277,6 +277,10 @@ async def __chat(self, messages, model, temperature, **kwargs):
             yield choice, None
             return
         async for chunk in res:
+            if not chunk.choices:
+                logger.error(f"OpenAI error: {chunk}")
+                yield None, None
+                return
             yield chunk.choices[0], None
 
     def get_models(self):

diff --git a/app/sync.py b/app/sync.py
@@ -27,10 +27,13 @@ async def get_sync_data(request: Request, after: str = Query(None)):
             data = json.loads(f.read())
         if after:
             after_time = datetime.fromisoformat(after.replace("Z", "+00:00"))
+            logger.info(f"Filtering data after {after_time}")
             data["updated"] = [
                 item
                 for item in data["updated"]
-                if datetime.fromisoformat(item["updated_at"].replace("Z", "+00:00"))
+                if datetime.fromisoformat(
+                    item["client_updated_at"].replace("Z", "+00:00")
+                )
                 > after_time
             ]
         return data