cbrkit 0.28.1__tar.gz → 0.28.2__tar.gz

{cbrkit-0.28.1 → cbrkit-0.28.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: cbrkit
-Version: 0.28.1
+Version: 0.28.2
 Summary: Customizable Case-Based Reasoning (CBR) toolkit for Python with a built-in API and CLI
 Keywords: cbr,case-based reasoning,api,similarity,nlp,retrieval,cli,tool,library
 Author: Mirko Lenz

{cbrkit-0.28.1 → cbrkit-0.28.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cbrkit"
-version = "0.28.1"
+version = "0.28.2"
 description = "Customizable Case-Based Reasoning (CBR) toolkit for Python with a built-in API and CLI"
 authors = [{ name = "Mirko Lenz", email = "mirko@mirkolenz.com" }]
 readme = "README.md"

{cbrkit-0.28.1 → cbrkit-0.28.2}/src/cbrkit/system.py

@@ -17,18 +17,17 @@ __all__ = [
 
 
 @dataclass(slots=True, frozen=True)
-class System[K: str | int, V: BaseModel, S: Float]:
+class System[K: str | int, V: BaseModel, S: Float, P: str]:
     casebase: cbrkit.typing.Casebase[K, V]
-    model: type[V]
     retriever_pipelines: Mapping[
-        str, MaybeSequence[cbrkit.typing.RetrieverFunc[K, V, S]]
+        P, MaybeSequence[cbrkit.typing.RetrieverFunc[K, V, S]]
     ] = field(default_factory=dict)
-    reuser_pipelines: Mapping[str, MaybeSequence[cbrkit.typing.ReuserFunc[K, V, S]]] = (
+    reuser_pipelines: Mapping[P, MaybeSequence[cbrkit.typing.ReuserFunc[K, V, S]]] = (
         field(default_factory=dict)
     )
 
     def get_retriever_pipeline(
-        self, name: str, limit: int | None
+        self, name: P, limit: int | None
     ) -> Sequence[cbrkit.typing.RetrieverFunc[K, V, S]]:
         retrievers = produce_sequence(self.retriever_pipelines[name])
 
@@ -43,7 +42,7 @@ class System[K: str | int, V: BaseModel, S: Float]:
     def retrieve(
         self,
         query: V,
-        retriever_pipeline: str,
+        retriever_pipeline: P,
         limit: int | None = None,
     ) -> cbrkit.retrieval.QueryResultStep[K, V, S]:
         return cbrkit.retrieval.apply_query(
@@ -55,7 +54,7 @@ class System[K: str | int, V: BaseModel, S: Float]:
     def reuse(
         self,
         query: V,
-        reuser_pipeline: str,
+        reuser_pipeline: P,
     ) -> cbrkit.retrieval.QueryResultStep[K, V, S]:
         return cbrkit.reuse.apply_query(
             self.casebase,
@@ -66,8 +65,8 @@ class System[K: str | int, V: BaseModel, S: Float]:
     def cycle(
         self,
         query: V,
-        retriever_pipeline: str,
-        reuser_pipeline: str,
+        retriever_pipeline: P,
+        reuser_pipeline: P,
         limit: int | None = None,
     ) -> cbrkit.retrieval.QueryResultStep[K, V, S]:
         return cbrkit.cycle.apply_query(
@@ -117,9 +116,7 @@ class System[K: str | int, V: BaseModel, S: Float]:
 with cbrkit.helpers.optional_dependencies():
     from fastapi import FastAPI
 
-    def to_fastapi(system: System) -> FastAPI:
-        app = FastAPI()
-
+    def to_fastapi(system: System, app: FastAPI) -> FastAPI:
         for value in system.tools:
             app.post(f"/tool/{value.__name__}")(value)
 
@@ -135,9 +132,7 @@ with cbrkit.helpers.optional_dependencies():
 with cbrkit.helpers.optional_dependencies():
     from fastmcp import FastMCP
 
-    def to_fastmcp(system: System) -> FastMCP[Any]:
-        app = FastMCP()
-
+    def to_fastmcp[T](system: System, app: FastMCP[T]) -> FastMCP[T]:
         for value in system.tools:
             app.tool(value)
 
@@ -153,5 +148,5 @@ with cbrkit.helpers.optional_dependencies():
 with cbrkit.helpers.optional_dependencies():
     from pydantic_ai.toolsets import FunctionToolset
 
-    def to_pydantic_ai(system: System) -> FunctionToolset[Any]:
+    def to_pydantic_ai(system: System) -> FunctionToolset[None]:
         return FunctionToolset(system.tools)