h-ai-brain 0.0.16__tar.gz → 0.0.19__tar.gz

{h_ai_brain-0.0.16/src/h_ai_brain.egg-info → h_ai_brain-0.0.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h_ai_brain
-Version: 0.0.16
+Version: 0.0.19
 Summary: AI Research agent API
 Author-email: shoebill <shoebill.hai@gmail.com>
 Classifier: Programming Language :: Python :: 3
@@ -10,7 +10,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 License-File: NOTICE.txt
-Requires-Dist: h_message_bus~=0.0.21
+Requires-Dist: h_message_bus~=0.0.22
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Dynamic: license-file

{h_ai_brain-0.0.16 → h_ai_brain-0.0.19}/pyproject.toml

@@ -4,7 +4,7 @@
 
    [project]
    name = "h_ai_brain"
-   version = "0.0.16"
+   version = "0.0.19"
    authors = [
        {name = "shoebill", email = "shoebill.hai@gmail.com"},
    ]
@@ -17,11 +17,14 @@
        "Operating System :: OS Independent",
    ]
    dependencies = [
-       "h_message_bus~=0.0.21",
+       "h_message_bus~=0.0.22",
    ]
 
    [project.optional-dependencies]
    dev = [
        "pytest",
        # Other development dependencies
-   ]
+   ]
+
+   [tool.setuptools.package-data]
+    "src.h_ai.resources" = ["*.json"]  # This will include all JSON files in the resources directory

h_ai_brain-0.0.19/src/h_ai/application/hai_service.py

@@ -0,0 +1,20 @@
+from h_message_bus import NatsPublisherAdapter
+
+from ..domain.llm_config import LLMConfig
+from ..infrastructure.llm.ollama.ollama_generate_repository import OllamaGenerateRepository
+
+
+class HaiService:
+    def __init__(self, nats_publisher_adapter: NatsPublisherAdapter, llm_config: LLMConfig):
+        self.nats_publisher_adapter = nats_publisher_adapter
+        self.llm_config = llm_config
+        self.llm_generate_repository = OllamaGenerateRepository(
+            self.llm_config.url,
+            self.llm_config.model_name,
+            temperature=self.llm_config.temperature,
+            max_tokens=self.llm_config.max_tokens)
+
+    def ask_question(self, question: str, system_prompt: str = None, max_tokens = None) -> str:
+        return self.llm_generate_repository.generate(question, system_prompt, max_tokens)
+
+

h_ai_brain-0.0.19/src/h_ai/domain/llm_config.py

@@ -0,0 +1,6 @@
+class LLMConfig:
+    def __init__(self, url: str, model_name: str, temperature: float = 0.6, max_tokens: int = 2500):
+        self.url = url
+        self.model_name = model_name
+        self.temperature = temperature
+        self.max_tokens = max_tokens

{h_ai_brain-0.0.16 → h_ai_brain-0.0.19}/src/h_ai/infrastructure/llm/ollama/ollama_generate_repository.py

@@ -2,20 +2,22 @@ import uuid
 
 import requests
 
+from ..llm_response_cleaner import clean_llm_response
 from ....domain.reasoning.llm_generate_respository import LlmGenerateRepository
 
 
 class OllamaGenerateRepository(LlmGenerateRepository):
 
-    def __init__(self, api_url: str, model_name: str, system_prompt: str = None, temperature: float = None, seed: int = None):
+    def __init__(self, api_url: str, model_name: str, system_prompt: str = None, temperature: float = None, seed: int = None, max_tokens: int = 5000):
         self.model_name = model_name
         self.system_prompt = system_prompt
         self.api_url = api_url
         self.temperature = temperature
         self.seed = seed
+        self.max_tokens = max_tokens
 
 
-    def generate(self, user_prompt: str, system_prompt: str = None, session_id: str = None) -> str|None:
+    def generate(self, user_prompt: str, system_prompt: str = None, session_id: str = None, max_tokens: int = None) -> str|None:
         url = f"{self.api_url}/generate"
         random_guid = uuid.uuid4()
         guid_str = str(random_guid)
@@ -26,16 +28,18 @@ class OllamaGenerateRepository(LlmGenerateRepository):
             "system": system_prompt,
             "stream": False,
             "session": guid_str,
-            "num_ctx": "5000",
-            "temperature": "0.6"
+            "num_ctx": f"{self.max_tokens}",
+            "temperature": f"{self.temperature}"
         }
 
         if session_id:
             payload["session"] = session_id
         if self.seed:
-            payload["seed"] = self.seed
+            payload["seed"] = f"{self.seed}"
         if self.temperature:
-            payload["temperature"] = self.temperature
+            payload["temperature"] = f"{self.temperature}"
+        if max_tokens:
+            payload["num_ctx"] = f"{max_tokens}"
 
         try:
             print(payload)

{h_ai_brain-0.0.16 → h_ai_brain-0.0.19}/src/h_ai/infrastructure/llm/prompt_loader.py

@@ -4,7 +4,7 @@ import os
 
 class PromptLoader:
     def __init__(self, file_path):
-        # Resolve relative path to absolute path
+        # Resolve a relative path to an absolute path
         absolute_file_path = os.path.abspath(file_path)
 
         with open(absolute_file_path, "r") as file:

{h_ai_brain-0.0.16 → h_ai_brain-0.0.19/src/h_ai_brain.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h_ai_brain
-Version: 0.0.16
+Version: 0.0.19
 Summary: AI Research agent API
 Author-email: shoebill <shoebill.hai@gmail.com>
 Classifier: Programming Language :: Python :: 3
@@ -10,7 +10,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 License-File: NOTICE.txt
-Requires-Dist: h_message_bus~=0.0.21
+Requires-Dist: h_message_bus~=0.0.22
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Dynamic: license-file

{h_ai_brain-0.0.16 → h_ai_brain-0.0.19}/src/h_ai_brain.egg-info/SOURCES.txt

@@ -4,9 +4,8 @@ pyproject.toml
 src/h_ai/__init__.py
 src/h_ai/application/__init__.py
 src/h_ai/application/hai_service.py
-src/h_ai/application/system_prompts/__init__.py
-src/h_ai/application/system_prompts/roles/__init__.py
 src/h_ai/domain/__init__.py
+src/h_ai/domain/llm_config.py
 src/h_ai/domain/reasoning/__init__.py
 src/h_ai/domain/reasoning/llm_chat_repository.py
 src/h_ai/domain/reasoning/llm_generate_respository.py

h_ai_brain-0.0.19/src/h_ai_brain.egg-info/requires.txt

@@ -0,0 +1,4 @@
+h_message_bus~=0.0.22
+
+[dev]
+pytest

h_ai_brain-0.0.16/src/h_ai/application/hai_service.py

@@ -1,7 +0,0 @@
-from h_message_bus import NatsPublisherAdapter
-
-class HaiService:
-    def __init__(self, nats_publisher_adapter: NatsPublisherAdapter):
-        self.nats_publisher_adapter = nats_publisher_adapter
-
-

h_ai_brain-0.0.16/src/h_ai_brain.egg-info/requires.txt

@@ -1,4 +0,0 @@
-h_message_bus~=0.0.21
-
-[dev]
-pytest