corrected GetAIModel query in sdk

Author: psaesha

dataspace_sdk/resources/aimodels.py

@@ -86,8 +86,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
@@ -128,7 +128,6 @@ def get_by_id_graphql(self, model_id: str) -> Dict[str, Any]:
                 }
                 endpoints {
                     id
-                    name
                     url
                     httpMethod
                     authType

requirements.txt

@@ -69,6 +69,10 @@ zipp==3.18.1
 snapshot-selenium==0.0.2
 selenium==4.16.0  # Downgraded for compatibility
 
+psycopg2-binary
+tenacity
+torch
+
 # Security and Environment Management
 python-decouple==3.8
 django-debug-toolbar==4.2.0  # Downgraded for better Django compatibility