Merge pull request #20 from comfuture:typing-docmeta

Refactor Doc type handling to use DocMeta protocol

Author: comfuture

function_schema/core.py

@@ -12,7 +12,7 @@
     get_type_hints,
 )
 
-from .types import FunctionSchema, Doc
+from .types import FunctionSchema, Doc, DocMeta
 from .utils import unwrap_doc
 
 
@@ -102,7 +102,7 @@ def get_function_schema(
             # find description in param_args tuple
             try:
                 description = next(
-                    unwrap_doc(arg) for arg in param_args if isinstance(arg, Doc)
+                    unwrap_doc(arg) for arg in param_args if isinstance(arg, DocMeta)
                 )
             except StopIteration:
                 try:

function_schema/types.py

@@ -15,14 +15,20 @@
         from typing_extensions import Doc
     except ImportError:
 
-        @runtime_checkable
-        class Doc(Protocol):
+        class Doc:
             documentation: str
 
             def __init__(self, documentation: str, /):
                 self.documentation = documentation
 
 
+@runtime_checkable
+class DocMeta(Protocol):
+    """Represents the protocol for the Doc class."""
+
+    documentation: str
+
+
 class ParamSchema(TypedDict):
     """
     Represents the schema for a parameter.