naas-abi-core 1.0.2__tar.gz → 1.0.3__tar.gz

{naas_abi_core-1.0.2 → naas_abi_core-1.0.3}/=0.9.13

@@ -1,6 +1,6 @@
 Collecting uv
   Downloading uv-0.9.14-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (11 kB)
 Downloading uv-0.9.14-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (21.7 MB)
-   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 21.7/21.7 MB 236.3 MB/s  0:00:00
+   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 21.7/21.7 MB 334.0 MB/s  0:00:00
 Installing collected packages: uv
 Successfully installed uv-0.9.14

{naas_abi_core-1.0.2 → naas_abi_core-1.0.3}/CHANGELOG.md

@@ -2,6 +2,14 @@
 
 <!-- version list -->
 
+## v1.0.3 (2025-12-01)
+
+### Bug Fixes
+
+- Working on fixing mypy check for core and naas-abi-marketplace linkedin
+  ([`fc11437`](https://github.com/jupyter-naas/abi/commit/fc11437fa5b655805327783016318416d17a0f1b))
+
+
 ## v1.0.2 (2025-12-01)
 
 ### Bug Fixes

{naas_abi_core-1.0.2 → naas_abi_core-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: naas-abi-core
-Version: 1.0.2
+Version: 1.0.3
 Summary: Abi framework allowing you to build your AI system.
 Author-email: Maxime Jublou <maxime@naas.ai>, Florent Ravenel <florent@naas.ai>, Jeremy Ravenel <jeremy@naas.ai>
 Requires-Python: <4,>=3.10

{naas_abi_core-1.0.2 → naas_abi_core-1.0.3}/naas_abi_core/module/Module.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import glob
 import os
-from typing import Dict, List
+from typing import Dict, List, cast
 
 from naas_abi_core import logger
 from naas_abi_core.engine.engine_configuration.EngineConfiguration import GlobalConfig
@@ -13,6 +13,7 @@ from naas_abi_core.pipeline.pipeline import Pipeline
 from naas_abi_core.services.agent.Agent import Agent
 from naas_abi_core.workflow.workflow import Workflow
 from pydantic import BaseModel, ConfigDict
+from typing_extensions import Generic, Self, TypeVar
 
 
 class ModuleDependencies:
@@ -46,13 +47,16 @@ class ModuleConfiguration(BaseModel):
     global_config: GlobalConfig
 
 
-class BaseModule:
+TConfig = TypeVar("TConfig", bound=ModuleConfiguration)
+
+
+class BaseModule(Generic[TConfig]):
     """Base interface class for ABI modules."""
 
-    _instances: Dict[type, "BaseModule"] = {}
+    _instances: Dict[type, Self] = {}
 
     _engine: EngineProxy
-    _configuration: ModuleConfiguration
+    _configuration: TConfig
     dependencies: ModuleDependencies = ModuleDependencies(modules=[], services=[])
 
     __ontologies: List[str] = []
@@ -61,7 +65,7 @@ class BaseModule:
     __workflows: List[Workflow] = []
     __pipelines: List[Pipeline] = []
 
-    def __init__(self, engine: EngineProxy, configuration: ModuleConfiguration):
+    def __init__(self, engine: EngineProxy, configuration: TConfig):
         assert isinstance(configuration, ModuleConfiguration), (
             "configuration must be an instance of ModuleConfiguration"
         )
@@ -76,7 +80,9 @@ class BaseModule:
             "BaseModule.Configuration must be a subclass of ModuleConfiguration"
         )
 
-        self._instances[self.__class__] = self
+        self_instance: Self = cast(Self, self)
+
+        self._instances[self.__class__] = self_instance
 
     @classmethod
     def get_dependencies(cls) -> List[str]:
@@ -84,7 +90,7 @@ class BaseModule:
         return getattr(cls, "dependencies", [])
 
     @classmethod
-    def get_instance(cls) -> "BaseModule":
+    def get_instance(cls) -> Self:
         if cls not in cls._instances:
             raise ValueError(f"Module {cls} not initialized")
         return cls._instances[cls]
@@ -94,7 +100,7 @@ class BaseModule:
         return self._engine
 
     @property
-    def configuration(self) -> ModuleConfiguration:
+    def configuration(self) -> TConfig:
         return self._configuration
 
     @property
@@ -243,3 +249,4 @@ class BaseModule:
 #     def on_initialized(self):
 #         if hasattr(self.imported_module, "on_initialized"):
 #             self.imported_module.on_initialized()
+#             self.imported_module.on_initialized()

{naas_abi_core-1.0.2 → naas_abi_core-1.0.3}/naas_abi_core/utils/Expose.py

@@ -4,8 +4,9 @@ from abc import ABC, abstractmethod
 from enum import Enum
 from typing import TYPE_CHECKING
 
-if TYPE_CHECKING:
-    from fastapi import APIRouter
+from fastapi import APIRouter
+
+if TYPE_CHECKING:    
     from langchain_core.tools import BaseTool
 
 
@@ -51,3 +52,4 @@ class Expose(ABC):
             NotImplementedError: If the concrete class does not implement this method
         """
         raise NotImplementedError()
+        raise NotImplementedError()

{naas_abi_core-1.0.2 → naas_abi_core-1.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "naas-abi-core"
-version = "1.0.2"
+version = "1.0.3"
 description = "Abi framework allowing you to build your AI system."
 authors = [{ name = "Maxime Jublou", email = "maxime@naas.ai" },{ name = "Florent Ravenel", email = "florent@naas.ai" }, { name = "Jeremy Ravenel", email = "jeremy@naas.ai" }]
 requires-python = ">=3.10,<4"