fix dataset query in sdk

Author: Deepthi-Chand

dataspace_sdk/resources/datasets.py

@@ -293,7 +293,7 @@ def get_prompt_by_id(self, dataset_id: str) -> Dict[str, Any]:
         """
         query = """
         query GetPromptDataset($id: UUID!) {
-            dataset(id: $id) {
+            get_dataset(id: $id) {
                 id
                 title
                 description
@@ -352,7 +352,7 @@ def get_prompt_by_id(self, dataset_id: str) -> Dict[str, Any]:
 
             raise DataSpaceAPIError(f"GraphQL error: {response['errors']}")
 
-        result: Dict[str, Any] = response.get("data", {}).get("dataset", {})
+        result: Dict[str, Any] = response.get("data", {}).get("get_dataset", {})
         return result
 
     def list_prompts(