fix: ensure that failed project doesn't cause downstream issues

Author: joschkabraun

parea/client.py

@@ -77,29 +77,23 @@ def __attrs_post_init__(self):
         parea_logger.set_client(self._client)
 
         if self.api_key:
-            try:
-                project_api_response: CreateGetProjectResponseSchema = self._create_or_get_project(self.project_name)
-                if project_api_response.was_created:
-                    print(f"Created project {project_api_response.name}")
-                self._project = structure(asdict(project_api_response), ProjectSchema)
-                parea_logger.set_project_uuid(self._project.uuid, self.project_name)
-            except httpx.HTTPStatusError as e:
-                if e.response.status_code == 502:
-                    logger.error("Error creating Parea project please try again")
-                else:
-                    raise
+            self._get_project_uuid(reraise_any_exception=False)
         else:
             logger.warning("No API key found. Parea client will not be able to send data to the Parea API.")
 
-    def _get_project_uuid(self) -> Optional[str]:
-        if not (self._project and self._project.uuid):
-            project_api_response: CreateGetProjectResponseSchema = self._create_or_get_project(self.project_name or "default")
-            self._project = structure(asdict(project_api_response), ProjectSchema)
-            parea_logger.set_project_uuid(self._project.uuid, self.project_name)
+    def _get_project_uuid(self, reraise_any_exception: bool = True) -> Optional[str]:
         try:
+            if not (self._project and self._project.uuid):
+                project_api_response: CreateGetProjectResponseSchema = self._create_or_get_project(self.project_name or "default")
+                if project_api_response.was_created:
+                    print(f"Created project {project_api_response.name}")
+                self._project = structure(asdict(project_api_response), ProjectSchema)
+                parea_logger.set_project_uuid(self._project.uuid, self.project_name)
             return self._project.uuid
         except Exception as e:
-            logger.error(f"Parea: Error getting project uuid for project {self.project_name}: {e}")
+            logger.error(f"Parea: Error getting project UUID for project {self.project_name}: {e}")
+            if reraise_any_exception:
+                raise
             return None
 
     def wrap_openai_client(self, client: "OpenAI", integration: Optional[str] = None) -> None:

parea/parea_logger.py

@@ -34,14 +34,14 @@ def set_project_uuid(self, project_uuid: str, project_name: str) -> None:
         self._project_uuid = project_uuid
         self._project_name = project_name
 
-    def _get_project_uuid(self) -> Optional[str]:
-        if not self._project_uuid:
-            self._project_uuid = self._create_or_get_project(self._project_name or "default").uuid
+    def _get_project_uuid(self) -> str:
         try:
+            if not self._project_uuid:
+                self._project_uuid = self._create_or_get_project(self._project_name or "default").uuid
             return self._project_uuid
         except Exception as e:
             logger.error(f"PareaLogger: Error getting project uuid for project {self._project_name}: {e}")
-            return None
+            raise
 
     def _create_or_get_project(self, name: str) -> CreateGetProjectResponseSchema:
         r = self._client.request(
@@ -70,7 +70,7 @@ def record_log(self, data: TraceLog) -> None:
 
     async def arecord_log(self, data: TraceLog) -> None:
         data = serialize_metadata_values(data)
-        data.project_uuid = self._project_uuid
+        data.project_uuid = self._get_project_uuid()
         await self._client.request_async(
             "POST",
             LOG_ENDPOINT,