Merge pull request #1048 from parea-ai/PAI-1462-ensure-that-failed-parea-project-creation-doesnt-cause-any-issues

Pai 1462 ensure that failed parea project creation doesnt cause any issues

Author: joschkabraun

parea/client.py

@@ -76,31 +76,21 @@ def __attrs_post_init__(self):
         self._client.set_api_key(self.api_key)
         parea_logger.set_client(self._client)
 
-        if self.api_key:
-            try:
-                project_api_response: CreateGetProjectResponseSchema = self._create_or_get_project(self.project_name)
+        if not self.api_key:
+            logger.warning("No API key found. Parea client will not be able to send data to the Parea API.")
+
+    def _get_project_uuid(self) -> 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)
-            except httpx.HTTPStatusError as e:
-                if e.response.status_code == 502:
-                    logger.error("Error creating Parea project please try again")
-                else:
-                    raise
-        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)
-        try:
             return self._project.uuid
         except Exception as e:
-            logger.error(f"Parea: Error getting project uuid for project {self.project_name}: {e}")
-            return None
+            logger.error(f"Parea: Error getting project UUID for project {self.project_name}: {e}")
+            raise
 
     def wrap_openai_client(self, client: "OpenAI", integration: Optional[str] = None) -> None:
         """Only necessary for instance client with OpenAI version >= 1.0.0"""

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,

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 [tool.poetry]
 name = "parea-ai"
 packages = [{ include = "parea" }]
-version = "0.2.200"
+version = "0.2.201"
 description = "Parea python sdk"
 readme = "README.md"
 authors = ["joel-parea-ai <[email protected]>"]