diff --git a/basalt/_version.py b/basalt/_version.py
index 3d18726..dd9b22c 100644
--- a/basalt/_version.py
+++ b/basalt/_version.py
@@ -1 +1 @@
-__version__ = "0.5.0"
+__version__ = "0.5.1"
diff --git a/basalt/sdk/promptsdk.py b/basalt/sdk/promptsdk.py
index 5a0d9fd..ea99b38 100644
--- a/basalt/sdk/promptsdk.py
+++ b/basalt/sdk/promptsdk.py
@@ -1,8 +1,8 @@
 from typing import Optional, Dict, Tuple, cast
 
-from ..ressources.monitor.generation_types import GenerationParams, PromptReference
 from ..ressources.monitor.trace_types import TraceParams
 from ..ressources.prompts.prompt_types import Prompt as IPrompt, PromptParams
+from ..utils.protocols import IPromptSDK
 from ..utils.dtos import (
 	GetPromptDTO,
 	GetPromptResult,
@@ -28,7 +28,7 @@
 from ..utils.flusher import Flusher
 from datetime import datetime
 
-class PromptSDK:
+class PromptSDK(IPromptSDK):
     """
     SDK for interacting with Basalt prompts.
     """
@@ -50,8 +50,8 @@ def __init__(
     async def get(
         self,
         slug: str,
-        version: Optional[str] = None,
         tag: Optional[str] = None,
+        version: Optional[str] = None,
         variables: Optional[Dict[str, str]] = None,
         cache_enabled: bool = True
     ) -> GetPromptResult:
@@ -111,8 +111,8 @@ async def get(
     def get_sync(
         self,
         slug: str,
-        version: Optional[str] = None,
         tag: Optional[str] = None,
+        version: Optional[str] = None,
         variables: Optional[Dict[str, str]] = None,
         cache_enabled: bool = True
     ) -> GetPromptResult:
@@ -210,8 +210,8 @@ def _prepare_monitoring(self,prompt: IPrompt,
     async def describe(
         self,
         slug: str,
-        version: Optional[str] = None,
         tag: Optional[str] = None,
+        version: Optional[str] = None,
     ) -> DescribeResult:
         """
         Get details about a prompt by slug, optionally specifying version and tag.
@@ -250,8 +250,8 @@ async def describe(
     def describe_sync(
         self,
         slug: str,
-        version: Optional[str] = None,
         tag: Optional[str] = None,
+        version: Optional[str] = None,
     ) -> DescribeResult:
         """
         Synchronously get details about a prompt by slug, optionally specifying version and tag.