naas-abi 1.0.0__py3-none-any.whl

naas_abi/mappings.py

@@ -0,0 +1,83 @@
+COLORS_NODES = {
+    # Role
+    "http://ontology.naas.ai/abi/Offering": "#f61685",
+    "http://ontology.naas.ai/abi/Product": "#C4116A",
+    "http://ontology.naas.ai/abi/Service": "#F973B5",
+    "http://ontology.naas.ai/abi/Positioning": "#FDD0E6",
+    # Capability
+    "http://ontology.naas.ai/abi/HumanCapabilities": "#D397F8",
+    "http://ontology.naas.ai/abi/TechnologicalCapabilities": "#bd87df",
+    "https://www.commoncoreontologies.org/ont00000089": "#bd87df",  # Skill
+    # Temporal Region
+    "https://www.commoncoreontologies.org/ont00000832": "#47DD82",
+    "http://ontology.naas.ai/abi/ISO8601UTCDateTime": "#47DD82",
+    # Agent Ontology
+    'https://www.commoncoreontologies.org/ont00001180': 'white', # Organization
+    'https://www.commoncoreontologies.org/ont00000443': 'white', # Commercial Organization
+    'http://ontology.naas.ai/abi/StockMarket': 'white', # Stock Market Institution
+    'https://www.commoncoreontologies.org/ont00001302': 'white', # Civil Organization
+    'https://www.commoncoreontologies.org/ont00000564': 'white', # Educational Organization
+    'https://www.commoncoreontologies.org/ont00000408': 'white', # Government Organization
+    'https://www.commoncoreontologies.org/ont00001239': '#f6f6f6', # Group Organization
+    'http://ontology.naas.ai/abi/Industry': '#f6f6f6', # Group of organizations
+    'http://ontology.naas.ai/abi/LinkedInIndustry': '#f6f6f6', # Group of organizations
+    'https://www.commoncoreontologies.org/ont00001262': '#919191', # Group
+    'https://www.commoncoreontologies.org/ont00000647': '#919191', # Member
+    'https://www.commoncoreontologies.org/ont00000914': 'white', # Group Member
+    'https://www.commoncoreontologies.org/ont00000175': '#3e3e3e', # Member Role
+    'http://ontology.naas.ai/abi/leadership#Position': '#905e0e', # Position
+    # Event Ontology
+    "http://ontology.naas.ai/abi/ActOfPartnership": "black",
+    "http://ontology.naas.ai/abi/ActOfJointVenture": "black",
+    "http://ontology.naas.ai/abi/ActOfMarketingAlliance": "black",
+    "http://ontology.naas.ai/abi/ActOfResearchCollaboration": "black",
+    "http://ontology.naas.ai/abi/ActOfTechnologyLicensing": "black",
+    "http://ontology.naas.ai/abi/ActOfDistributionAgreement": "black",
+    "http://ontology.naas.ai/abi/ActOfOrganizationalMerger": "black",
+    "http://ontology.naas.ai/abi/ActOfOrganizationalAcquisition": "black",
+    "http://ontology.naas.ai/abi/ActOfSubsidiaryEstablishment": "black",
+    # Information Content
+    "https://www.commoncoreontologies.org/ont00001331": "#AECCE4",  # Legal Name
+    "http://ontology.naas.ai/abi/OrganizationSize": "#9ABDDC",
+    "http://ontology.naas.ai/abi/Ticker": "#6C849A",
+    "http://ontology.naas.ai/abi/Address": "#4D5F6E",
+    "http://ontology.naas.ai/abi/Market": "#4169E1",
+    "http://ontology.naas.ai/abi/MarketSegment": "#6495ED",
+    "http://ontology.naas.ai/abi/LinkedInProfilePage": "#0A66C2",
+    "http://ontology.naas.ai/abi/IncomeStatement": "#847c35",
+    "http://ontology.naas.ai/abi/BalanceSheet": "#9d965d",
+    "http://ontology.naas.ai/abi/CashFlowStatement": "#c1bd9a",
+    # Finance Ontology
+    "http://ontology.naas.ai/abi/Revenue": "#bf9b30",
+    "http://ontology.naas.ai/abi/Expenses": "#bf9b30",
+    "http://ontology.naas.ai/abi/Profit": "#bf9b30",
+    "http://ontology.naas.ai/abi/EBITDA": "#bf9b30",
+    "http://ontology.naas.ai/abi/KPI": "#d4bf79",
+    # Facility Ontology
+    "http://ontology.naas.ai/abi/GlobalHeadquarters": "#9D7153",
+    "http://ontology.naas.ai/abi/RegionalHeadquarters": "#b0917c",
+    # Geospatial Ontology
+    "http://ontology.naas.ai/abi/Country": "#8e7010",
+    "http://ontology.naas.ai/abi/CountryISO3166-1": "#8e7010",
+    "http://ontology.naas.ai/abi/State": "#c6a84f",
+    "https://www.commoncoreontologies.org/ont00000887": "#aa8b26",
+    # Euronext Index
+    "http://ontology.naas.ai/abi/DataSource": "#5b1f00",
+    "http://ontology.naas.ai/abi/DataSourceComponent": "#713112",
+    "http://ontology.naas.ai/abi/TradingLocation": "#09a12d",
+    "http://ontology.naas.ai/abi/euronext#Index": "#7fed0e",
+    # LinkedIn
+    "http://ontology.naas.ai/abi/linkedin#LinkedInProfile": "white",
+    "http://ontology.naas.ai/abi/linkedin#PositionGroup": "white",
+    "http://ontology.naas.ai/abi/linkedin#Position": "grey",
+    "http://ontology.naas.ai/abi/linkedin#Language": "#D397F8",
+    "http://ontology.naas.ai/abi/linkedin#Skill": "#bd87df",
+    "http://ontology.naas.ai/abi/linkedin#Certification": "#76b3ec",
+    "http://ontology.naas.ai/abi/linkedin#Course": "#b9986d",
+    
+    # "http://ontology.naas.ai/abi/linkedin#Skill": "white",
+    # "http://ontology.naas.ai/abi/linkedin#Education": "white",
+    # "http://ontology.naas.ai/abi/linkedin#Certification": "white",
+    # "http://ontology.naas.ai/abi/linkedin#Company": "white",
+    # "http://ontology.naas.ai/abi/linkedin#Industry": "white",
+}

naas_abi/models/airgap_gemma.py

@@ -0,0 +1,220 @@
+import json
+from typing import Any, Iterator, List, Optional
+
+import requests
+from langchain_core.callbacks.manager import CallbackManagerForLLMRun
+from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage, HumanMessage
+from langchain_core.outputs import ChatResult
+from langchain_core.outputs.chat_generation import ChatGenerationChunk
+from langchain_openai import ChatOpenAI
+from naas_abi_core import logger
+from naas_abi_core.models.Model import ChatModel
+
+
+class AirgapChatOpenAI(ChatOpenAI):
+    """Minimal wrapper for Docker Model Runner with basic tool support"""
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self._tools = []
+
+    def bind_tools(self, tools, **kwargs):
+        # Just return self without storing tools - we don't need complex tool handling
+        return self
+
+    def bind(self, **kwargs):
+        # Just return self - keep it simple
+        return self
+
+    @property
+    def _llm_type(self) -> str:
+        return "airgap_chat_openai"
+
+    def _generate(
+        self,
+        messages: List[BaseMessage],
+        stop: Optional[List[str]] = None,
+        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        **kwargs: Any,
+    ) -> ChatResult:
+        # Extract system prompt and user message with improved formatting
+        system_prompt = ""
+        user_msg: str | None = None
+
+        for msg in messages:
+            if hasattr(msg, "content") and msg.content:
+                if "SystemMessage" in str(type(msg)):
+                    if isinstance(msg.content, str):
+                        system_prompt += msg.content + "\n"
+                elif isinstance(msg, HumanMessage):
+                    assert isinstance(msg.content, str)
+                    user_msg = msg.content
+
+        # Always ensure we have a valid user message
+        if not user_msg or (isinstance(user_msg, str) and not user_msg.strip()):
+            user_msg = "Hello"
+        elif not isinstance(user_msg, str):
+            user_msg = str(user_msg)
+
+        # Build GPT-style prompt with clear instruction formatting
+        if system_prompt.strip():
+            prompt = f"System: {system_prompt.strip()}\n\n"
+        else:
+            prompt = ""
+
+        prompt += f"User: {user_msg}\n\nAssistant:"
+        messages = [HumanMessage(content=prompt)]
+
+        # Clean kwargs
+        clean_kwargs = {
+            k: v
+            for k, v in kwargs.items()
+            if k in ["temperature", "max_tokens", "stop"]
+        }
+
+        # Get response
+        result = super()._generate(
+            messages, stop=stop, run_manager=run_manager, **clean_kwargs
+        )
+
+        # Simple tool call handling - just ensure we have proper AIMessage format
+        if result.generations:
+            content = result.generations[0].message.content
+
+            # Always create AIMessage with empty tool_calls to prevent routing issues
+            ai_message = AIMessage(content=content, additional_kwargs={})
+            ai_message.tool_calls = []
+            result.generations[0].message = ai_message
+
+        return result
+
+    def _stream(
+        self,
+        messages: List[BaseMessage],
+        stop: Optional[List[str]] = None,
+        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        **kwargs: Any,
+    ) -> Iterator[ChatGenerationChunk]:
+        """Stream tokens from the Docker Model Runner"""
+        # Extract system prompt and user message with improved formatting
+        system_prompt = ""
+        user_msg: str | None = None
+
+        for msg in messages:
+            if hasattr(msg, "content") and msg.content:
+                if "SystemMessage" in str(type(msg)):
+                    if isinstance(msg.content, str):
+                        system_prompt += msg.content + "\n"
+                elif isinstance(msg, HumanMessage):
+                    assert isinstance(msg.content, str)
+                    user_msg = msg.content
+
+        # Always ensure we have a valid user message
+        if not user_msg or (isinstance(user_msg, str) and not user_msg.strip()):
+            user_msg = "Hello"
+        elif not isinstance(user_msg, str):
+            user_msg = str(user_msg)
+
+        # Build GPT-style prompt with clear instruction formatting
+        if system_prompt.strip():
+            prompt = f"System: {system_prompt.strip()}\n\n"
+        else:
+            prompt = ""
+
+        prompt += f"User: {user_msg}\n\nAssistant:"
+
+        # Make streaming request to Docker Model Runner
+        try:
+            response = requests.post(
+                f"{self.openai_api_base}/chat/completions",
+                json={
+                    "model": self.model_name,
+                    "messages": [{"role": "user", "content": prompt}],
+                    "stream": True,
+                    "temperature": kwargs.get("temperature", self.temperature),
+                    "max_tokens": kwargs.get("max_tokens", 512),
+                },
+                stream=True,
+                timeout=30,
+            )
+            response.raise_for_status()
+
+            # Process streaming response
+            for line in response.iter_lines():
+                if line:
+                    line = line.decode("utf-8")
+                    if line.startswith("data: "):
+                        data = line[6:]  # Remove 'data: ' prefix
+                        if data.strip() == "[DONE]":
+                            break
+                        try:
+                            chunk_data = json.loads(data)
+                            if (
+                                "choices" in chunk_data
+                                and len(chunk_data["choices"]) > 0
+                            ):
+                                choice = chunk_data["choices"][0]
+                                if "delta" in choice and "content" in choice["delta"]:
+                                    content = choice["delta"]["content"]
+                                    if content:
+                                        yield ChatGenerationChunk(
+                                            message=AIMessageChunk(content=content),
+                                            generation_info={
+                                                "finish_reason": choice.get(
+                                                    "finish_reason"
+                                                )
+                                            },
+                                        )
+                        except json.JSONDecodeError:
+                            continue
+
+        except requests.exceptions.Timeout:
+            logger.error("Docker Model Runner timeout - model may be overloaded")
+            yield ChatGenerationChunk(
+                message=AIMessageChunk(
+                    content="⚠️ Model response timeout. Try a shorter message or switch to cloud mode."
+                ),
+                generation_info={"finish_reason": "timeout"},
+            )
+        except requests.exceptions.ConnectionError:
+            logger.error("Docker Model Runner connection failed")
+            yield ChatGenerationChunk(
+                message=AIMessageChunk(
+                    content="❌ Local model unavailable. Use 'make model-up' or switch to cloud mode."
+                ),
+                generation_info={"finish_reason": "connection_error"},
+            )
+        except Exception as e:
+            logger.error(f"Docker Model Runner error: {e}")
+            yield ChatGenerationChunk(
+                message=AIMessageChunk(
+                    content="🔄 Model error. Try restarting with 'make model-down && make model-up'"
+                ),
+                generation_info={"finish_reason": "error"},
+            )
+
+
+MODEL_ID = "ai/gemma3"
+NAME = "gemma3-airgap"
+DESCRIPTION = (
+    "Gemma3 model running in airgap mode via Docker Model Runner with tool support."
+)
+IMAGE = "https://naasai-public.s3.eu-west-3.amazonaws.com/abi-demo/ontology_ABI.png"
+CONTEXT_WINDOW = 8192
+PROVIDER = "google"
+
+model: ChatModel = ChatModel(
+    model_id=MODEL_ID,
+    provider=PROVIDER,
+    name=NAME,
+    description=DESCRIPTION,
+    image=IMAGE,
+    model=AirgapChatOpenAI(
+        model=MODEL_ID,
+        temperature=0.2,  # Even lower temperature for faster, more focused responses
+        max_tokens=512,  # Shorter responses for speed
+        openai_api_base="http://localhost:12434/engines/v1",
+        api_key="ignored",
+    ),
+    context_window=CONTEXT_WINDOW,
+)

naas_abi/models/airgap_qwen.py

@@ -0,0 +1,24 @@
+from langchain_openai import ChatOpenAI
+from naas_abi_core.models.Model import ChatModel
+
+MODEL_ID = "ai/qwen3"
+PROVIDER = "qwen"
+NAME = "qwen3-airgap"
+DESCRIPTION = (
+    "Qwen3 model running in airgap mode via Docker Model Runner with tool support."
+)
+IMAGE = "https://naasai-public.s3.eu-west-3.amazonaws.com/abi-demo/ontology_ABI.png"
+CONTEXT_WINDOW = 8192
+
+model: ChatModel = ChatModel(
+    model_id=MODEL_ID,
+    provider=PROVIDER,
+    name=NAME,
+    description=DESCRIPTION,
+    image=IMAGE,
+    model=ChatOpenAI(
+        model="ai/qwen3",  # Qwen3 8B - better performance with 16GB RAM
+        temperature=0.7,
+        base_url="http://localhost:12434/engines/v1",
+    ),
+)

naas_abi/models/default.py

@@ -0,0 +1,23 @@
+from typing import Literal
+
+from naas_abi import ABIModule
+
+
+def get_model():
+    ai_mode: Literal["cloud", "local", "airgap"] = (
+        ABIModule.get_instance().configuration.global_config.ai_mode
+    )
+    if (
+        ai_mode == "airgap"
+        or not ABIModule.get_instance().configuration.openai_api_key
+        and not ABIModule.get_instance().configuration.openrouter_api_key
+    ):
+        from naas_abi.models.airgap_qwen import model as airgap_model
+
+        return airgap_model
+    else:
+        from naas_abi_marketplace.ai.chatgpt.models.gpt_4_1_mini import (
+            model as cloud_model,
+        )
+
+        return cloud_model

naas_abi/models/gpt_4_1.py

@@ -0,0 +1,25 @@
+from langchain_openai import ChatOpenAI
+from naas_abi import ABIModule
+from naas_abi_core.models.Model import ChatModel
+from pydantic import SecretStr
+
+ID = "gpt-4.1"
+NAME = "gpt-4.1"
+DESCRIPTION = "GPT-4.1 excels at instruction following and tool calling, with broad knowledge across domains. It features a 1M token context window, and low latency without a reasoning step."
+IMAGE = "https://naasai-public.s3.eu-west-3.amazonaws.com/abi-demo/ontology_ABI.png"
+CONTEXT_WINDOW = 1047576
+OWNER = "openai"
+
+model: ChatModel = ChatModel(
+    model_id=ID,
+    name=NAME,
+    description=DESCRIPTION,
+    image=IMAGE,
+    owner=OWNER,
+    model=ChatOpenAI(
+        model=ID,
+        temperature=0,
+        api_key=SecretStr(ABIModule.get_instance().configuration.openai_api_key),
+    ),
+    context_window=CONTEXT_WINDOW,
+)