support auto fetch cozeloop space id

Author: yaozheng-fang

veadk/configs/tracing_configs.py

@@ -23,9 +23,11 @@
     DEFAULT_APMPLUS_OTEL_EXPORTER_ENDPOINT,
     DEFAULT_APMPLUS_OTEL_EXPORTER_SERVICE_NAME,
     DEFAULT_COZELOOP_OTEL_EXPORTER_ENDPOINT,
+    DEFAULT_COZELOOP_SPACE_NAME,
     DEFAULT_TLS_OTEL_EXPORTER_ENDPOINT,
     DEFAULT_TLS_OTEL_EXPORTER_REGION,
 )
+from veadk.integrations.ve_cozeloop.ve_cozeloop import VeCozeloop
 from veadk.integrations.ve_tls.ve_tls import VeTLS
 
 
@@ -58,27 +60,24 @@ class CozeloopConfig(BaseSettings):
         default="", alias="OBSERVABILITY_OPENTELEMETRY_COZELOOP_API_KEY"
     )
 
-    otel_exporter_space_id: str = Field(
-        default="", alias="OBSERVABILITY_OPENTELEMETRY_COZELOOP_SERVICE_NAME"
-    )
-
     # TODO: auto fetching via AK/SK pair
     # @cached_property
     # def otel_exporter_api_key(self) -> str:
     #     pass
 
-    # TODO: auto fetching workspace id
-    # @cached_property
-    # def otel_exporter_space_id(self) -> str:
-    #     workspace_id = os.getenv("OBSERVABILITY_OPENTELEMETRY_COZELOOP_SERVICE_NAME", "")
+    @cached_property
+    def otel_exporter_space_id(self) -> str:
+        workspace_id = os.getenv(
+            "OBSERVABILITY_OPENTELEMETRY_COZELOOP_SERVICE_NAME", ""
+        )
 
-    #     if not workspace_id:
-    #         # create a default one
-    #         workspace_id = VeCozeloop(self.otel_exporter_api_key).create_workspace(
-    #             workspace_name=DEFAULT_COZELOOP_SPACE_NAME
-    #         )
+        if not workspace_id:
+            # create a default one
+            workspace_id = VeCozeloop(self.otel_exporter_api_key).create_workspace(
+                workspace_name=DEFAULT_COZELOOP_SPACE_NAME
+            )
 
-    #     return workspace_id
+        return workspace_id
 
 
 class TLSConfig(BaseSettings):

veadk/integrations/ve_cozeloop/ve_cozeloop.py

@@ -31,30 +31,36 @@ def create_workspace(
             f"Automatically create Cozeloop workspace with name {workspace_name}"
         )
 
-        URL = "https://api.coze.cn/v1/workspaces"
+        try:
+            workspace_id = self.search_workspace_id(workspace_name=workspace_name)
+            logger.info(f"Get existing Cozeloop workspace ID: {workspace_id}")
 
-        headers = {
-            "Authorization": f"Bearer {self.api_key}",
-            "Content-Type": "application/json",
-        }
-
-        data = {
-            "name": workspace_name,
-            "description": "Created by Volcengine Agent Development Kit (VeADK)",
-        }
-
-        response = requests.post(URL, headers=headers, json=data)
-
-        if response.json().get("code") == 0:
-            workspace_id = response.json().get("data").get("id")
-            logger.info(f"Cozeloop workspace ID: {workspace_id}")
             return workspace_id
-        else:
-            raise Exception(
-                f"Failed to automatically create workspace: {response.json()}"
-            )
-
-    def get_workspace_id(
+        except Exception as _:
+            URL = "https://api.coze.cn/v1/workspaces"
+
+            headers = {
+                "Authorization": f"Bearer {self.api_key}",
+                "Content-Type": "application/json",
+            }
+
+            data = {
+                "name": workspace_name,
+                "description": "Created by Volcengine Agent Development Kit (VeADK)",
+            }
+
+            response = requests.post(URL, headers=headers, json=data)
+
+            if response.json().get("code") == 0:
+                workspace_id = response.json().get("data").get("id")
+                logger.info(f"New created Cozeloop workspace ID: {workspace_id}")
+                return workspace_id
+            else:
+                raise Exception(
+                    f"Failed to automatically create workspace: {response.json()}"
+                )
+
+    def search_workspace_id(
         self, workspace_name: str = DEFAULT_COZELOOP_SPACE_NAME
     ) -> str:
         logger.info(
@@ -73,7 +79,7 @@ def get_workspace_id(
             "page_size": 50,
         }
 
-        response = requests.post(URL, headers=headers, json=data)
+        response = requests.get(URL, headers=headers, json=data)
 
         if response.json().get("code") == 0:
             workspaces = response.json().get("data").get("workspaces", [])