eai_answers_lib 0.1.0__tar.gz

eai_answers_lib-0.1.0/PKG-INFO

@@ -0,0 +1,49 @@
+Metadata-Version: 2.4
+Name: eai_answers_lib
+Version: 0.1.0
+Summary: Library to interact with Expert AI Answers engine
+Author: Simone Martin
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+Requires-Dist: loguru>=0.7.3
+Requires-Dist: pydantic>=2.13.3
+Requires-Dist: requests>=2.33.1
+Provides-Extra: dev
+Requires-Dist: ruff>=0.9.0; extra == "dev"
+Requires-Dist: pytest>=8.0.0; extra == "dev"
+Requires-Dist: responses>=0.25.0; extra == "dev"
+
+# Answers Lib
+
+Library to interact with Expert AI Answers engine.
+
+## Installation
+
+```bash
+pip install eai-answers-lib
+```
+
+## Usage
+
+```python
+from eai_answers_lib import Answers
+
+a = Answers()
+result = a.get_answers(query="Your question")
+```
+
+## Authentication
+
+```python
+a = Answers(
+    enable_auth=True,
+    client_id="your-client-id",
+    client_secret="your-client-secret"
+)
+```
+
+## Environment Variables
+
+- `ANSWERS_ENDPOINT` - API endpoint URL
+- `ANSWERS_CLIENT_ID` - Client ID for authentication
+- `ANSWERS_CLIENT_SECRET` - Client secret for authentication

eai_answers_lib-0.1.0/README.md

@@ -0,0 +1,34 @@
+# Answers Lib
+
+Library to interact with Expert AI Answers engine.
+
+## Installation
+
+```bash
+pip install eai-answers-lib
+```
+
+## Usage
+
+```python
+from eai_answers_lib import Answers
+
+a = Answers()
+result = a.get_answers(query="Your question")
+```
+
+## Authentication
+
+```python
+a = Answers(
+    enable_auth=True,
+    client_id="your-client-id",
+    client_secret="your-client-secret"
+)
+```
+
+## Environment Variables
+
+- `ANSWERS_ENDPOINT` - API endpoint URL
+- `ANSWERS_CLIENT_ID` - Client ID for authentication
+- `ANSWERS_CLIENT_SECRET` - Client secret for authentication

eai_answers_lib-0.1.0/pyproject.toml

@@ -0,0 +1,40 @@
+[project]
+name = "eai_answers_lib"
+version = "0.1.0"
+description = "Library to interact with Expert AI Answers engine"
+authors = [{name = "Simone Martin"}]
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "loguru>=0.7.3",
+    "pydantic>=2.13.3",
+    "requests>=2.33.1",
+]
+
+[project.optional-dependencies]
+dev = [
+    "ruff>=0.9.0",
+    "pytest>=8.0.0",
+    "responses>=0.25.0",
+]
+
+[tool.uv]
+dev-dependencies = [
+    "ruff>=0.9.0",
+    "pytest>=8.0.0",
+    "responses>=0.25.0",
+]
+
+[tool.ruff]
+target-version = "py312"
+
+[tool.ruff.format]
+quote-style = "double"
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py", "*_test.py"]
+python_functions = ["test_*"]
+
+[tool.setuptools.packages.find]
+where = ["src"]

eai_answers_lib-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

eai_answers_lib-0.1.0/src/eai_answers_lib.egg-info/PKG-INFO

@@ -0,0 +1,49 @@
+Metadata-Version: 2.4
+Name: eai_answers_lib
+Version: 0.1.0
+Summary: Library to interact with Expert AI Answers engine
+Author: Simone Martin
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+Requires-Dist: loguru>=0.7.3
+Requires-Dist: pydantic>=2.13.3
+Requires-Dist: requests>=2.33.1
+Provides-Extra: dev
+Requires-Dist: ruff>=0.9.0; extra == "dev"
+Requires-Dist: pytest>=8.0.0; extra == "dev"
+Requires-Dist: responses>=0.25.0; extra == "dev"
+
+# Answers Lib
+
+Library to interact with Expert AI Answers engine.
+
+## Installation
+
+```bash
+pip install eai-answers-lib
+```
+
+## Usage
+
+```python
+from eai_answers_lib import Answers
+
+a = Answers()
+result = a.get_answers(query="Your question")
+```
+
+## Authentication
+
+```python
+a = Answers(
+    enable_auth=True,
+    client_id="your-client-id",
+    client_secret="your-client-secret"
+)
+```
+
+## Environment Variables
+
+- `ANSWERS_ENDPOINT` - API endpoint URL
+- `ANSWERS_CLIENT_ID` - Client ID for authentication
+- `ANSWERS_CLIENT_SECRET` - Client secret for authentication

eai_answers_lib-0.1.0/src/eai_answers_lib.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+pyproject.toml
+src/eai_answers_lib.egg-info/PKG-INFO
+src/eai_answers_lib.egg-info/SOURCES.txt
+src/eai_answers_lib.egg-info/dependency_links.txt
+src/eai_answers_lib.egg-info/requires.txt
+src/eai_answers_lib.egg-info/top_level.txt
+src/schemas/__init__.py
+src/schemas/answers_schemas.py
+src/services/__init__.py
+src/services/logger.py
+tests/test_answers.py

eai_answers_lib-0.1.0/src/eai_answers_lib.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

eai_answers_lib-0.1.0/src/eai_answers_lib.egg-info/requires.txt

@@ -0,0 +1,8 @@
+loguru>=0.7.3
+pydantic>=2.13.3
+requests>=2.33.1
+
+[dev]
+ruff>=0.9.0
+pytest>=8.0.0
+responses>=0.25.0

eai_answers_lib-0.1.0/src/eai_answers_lib.egg-info/top_level.txt

@@ -0,0 +1,2 @@
+schemas
+services

eai_answers_lib-0.1.0/src/schemas/answers_schemas.py

@@ -0,0 +1,35 @@
+
+from typing import Any, Dict, Optional, Union, List, Literal
+from pydantic import BaseModel
+
+
+    
+    
+class PackageAndCollection(BaseModel):
+    kb: str
+    description: str # collection description
+    id: str # collection id
+    
+class CollectionDocument(BaseModel):
+    data: str
+    dataAnnotation: Optional[str] = None
+    dataXml: Optional[str] = None
+    dataJson: Optional[Dict[Any, Any]] = None
+    size: Optional[int] = None
+    name: str
+
+class AddCollectionRequest(BaseModel):
+    cogitoAnalysisMode: Optional[str] = None
+    removeAnnotations: Optional[bool] = True
+    ocr: Optional[bool] = False
+    doubleColumn: Optional[bool] = False
+    pageImages: Optional[bool] = False
+    indexStrategies: List[str]
+    engine: str
+    analyzer: Optional[str] = None
+    status: Optional[str] = None
+    messages: Optional[List[Dict[str, str]]] = None
+    documents: List[CollectionDocument]
+    pkg: str
+    collection: str
+

eai_answers_lib-0.1.0/src/services/logger.py

@@ -0,0 +1,16 @@
+from pathlib import Path
+from loguru import logger
+
+logs_path = Path(".logs")
+logs_path.mkdir(parents=True, exist_ok=True)
+
+#logger.remove() # Remove the default console logger
+logger.add(
+    logs_path / "log_{time:YYYY-MM-DD}.log",  # File name with date
+    rotation="1 day",
+    retention="2 months",                # Keep 7 days of logs
+    # compression="zip",                 # Compress old logs
+    level="INFO",                      # Only log INFO and above
+    enqueue=True                       # Thread-safe logging
+)
+

eai_answers_lib-0.1.0/tests/test_answers.py

@@ -0,0 +1,156 @@
+import pytest
+import responses
+from src import Answers
+
+
+responses.mock.start()
+responses.mock.register_uri(
+    responses.POST,
+    "https://api.example.com/auth/realms/test-realm/protocol/openid-connect/token",
+    json={"access_token": "test-token", "expires_in": 3600},
+    status=200,
+)
+
+
+@pytest.fixture
+def mock_endpoint():
+    return "https://api.example.com"
+
+
+@pytest.fixture
+def answers(mock_endpoint):
+    return Answers(endpoint=mock_endpoint, enable_auth=False)
+
+
+@pytest.fixture
+def answers_with_auth(mock_endpoint):
+    return Answers(
+        endpoint=mock_endpoint,
+        enable_auth=True,
+        auth_path="auth",
+        realm="test-realm",
+        client_id="test-client",
+        client_secret="test-secret",
+    )
+
+
+class TestInit:
+    def test_init_success(self, mock_endpoint):
+        answers = Answers(endpoint=mock_endpoint)
+        assert answers.endpoint == mock_endpoint
+
+    def test_init_missing_endpoint_raises(self):
+        with pytest.raises(ValueError, match="endpoint must be provided"):
+            Answers(endpoint=None)
+
+    def test_init_with_auth(self, mock_endpoint):
+        answers = Answers(
+            endpoint=mock_endpoint,
+            enable_auth=True,
+            auth_path="auth",
+            realm="test-realm",
+            client_id="client",
+            client_secret="secret",
+        )
+        assert answers.enable_auth is True
+        assert answers.auth_path == "auth"
+        assert answers.realm == "test-realm"
+
+    def test_init_default_paths(self, mock_endpoint):
+        answers = Answers(endpoint=mock_endpoint)
+        assert answers.query_path == "v1/query"
+        assert answers.status_path == "v1/status"
+
+
+def test_refresh_bearer_success(answers_with_auth):
+    responses.mock.register_uri(
+        responses.POST,
+        "https://api.example.com/auth/realms/test-realm/protocol/openid-connect/token",
+        json={"access_token": "new-token", "expires_in": 3600},
+        status=200,
+    )
+    answers_with_auth.refresh_bearer()
+    assert answers_with_auth.bearer == "new-token"
+    assert answers_with_auth.token_expiration > 0
+
+
+def test_refresh_bearer_failure(answers_with_auth):
+    responses.mock.register_uri(
+        responses.POST,
+        "https://api.example.com/auth/realms/test-realm/protocol/openid-connect/token",
+        status=401,
+    )
+    with pytest.raises(Exception, match="Error Refreshing authentication Bearer"):
+        answers_with_auth.refresh_bearer()
+
+
+def test_get_auth_header_no_auth(answers):
+    headers = answers.get_auth_header()
+    assert headers == {}
+
+
+def test_get_auth_header_with_auth(answers_with_auth):
+    headers = answers_with_auth.get_auth_header()
+    assert headers == {"Authorization": "Bearer test-token"}
+
+
+@responses.activate
+def test_start_query_task_success(answers):
+    responses.post(
+        "https://api.example.com/v1/query/pkg/collection",
+        json={"success": True, "data": {"task_id": "task-123"}},
+        status=200,
+    )
+    task_id = answers.start_query_task(payload={}, headers={}, pkg="pkg", collection="collection")
+    assert task_id == "task-123"
+
+
+@responses.activate
+def test_start_query_task_failure(answers):
+    responses.post(
+        "https://api.example.com/v1/query/pkg/collection",
+        json={"success": False, "data": {"error": "failed"}},
+        status=200,
+    )
+    with pytest.raises(Exception):
+        answers.start_query_task(payload={}, headers={}, pkg="pkg", collection="collection")
+
+
+@responses.activate
+def test_poll_query_status_success(answers):
+    responses.get(
+        "https://api.example.com/v1/status/task-123",
+        json={"success": True, "data": {"completed": True, "query_tasks": ["t1"], "completed_tasks": ["t1"]}},
+        status=200,
+    )
+    status = answers.poll_query_status(task_id="task-123", headers={})
+    assert status["success"] is True
+    assert status["data"]["completed"] is True
+
+
+@responses.activate
+def test_poll_query_status_failure(answers):
+    responses.get(
+        "https://api.example.com/v1/status/task-123",
+        json={"success": False, "messages": ["error"]},
+        status=200,
+    )
+    status = answers.poll_query_status(task_id="task-123", headers={})
+    assert status["success"] is False
+
+
+@responses.activate
+def test_call_full_flow(answers):
+    responses.post(
+        "https://api.example.com/v1/query/pkg/collection",
+        json={"success": True, "data": {"task_id": "task-456"}},
+        status=200,
+    )
+    responses.get(
+        "https://api.example.com/v1/status/task-456",
+        json={"success": True, "data": {"completed": True}},
+        status=200,
+    )
+    result = answers(payload={}, pkg="pkg", collection="collection")
+    assert result["success"] is True
+    assert result["data"]["completed"] is True