fix aimodel endpoint in sdk

Author: Deepthi-Chand

dataspace_sdk/resources/aimodels.py

@@ -91,8 +91,8 @@ def get_by_id_graphql(self, model_id: str) -> Dict[str, Any]:
             Dictionary containing AI model information
         """
         query = """
-        query GetAIModel($id: UUID!) {
-            aiModel(id: $id) {
+        query GetAIModel($id: Int!) {
+            getAiModel(modelId: $id) {
                 id
                 name
                 displayName
@@ -171,7 +171,7 @@ def get_by_id_graphql(self, model_id: str) -> Dict[str, Any]:
             "/api/graphql",
             json_data={
                 "query": query,
-                "variables": {"id": model_id},
+                "variables": {"id": int(model_id)},
             },
         )
 
@@ -180,7 +180,7 @@ def get_by_id_graphql(self, model_id: str) -> Dict[str, Any]:
 
             raise DataSpaceAPIError(f"GraphQL error: {response['errors']}")
 
-        result: Dict[str, Any] = response.get("data", {}).get("aiModel", {})
+        result: Dict[str, Any] = response.get("data", {}).get("getAiModel", {})
         return result
 
     def list_all(