iatoolkit 2.15.2__tar.gz → 2.15.4__tar.gz

{iatoolkit-2.15.2/src/iatoolkit.egg-info → iatoolkit-2.15.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.15.2
+Version: 2.15.4
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-2.15.2 → iatoolkit-2.15.4}/src/iatoolkit/__init__.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-__version__ = "2.15.2"
+__version__ = "2.15.4"
 
 # Expose main classes and functions at the top level of the package
 

{iatoolkit-2.15.2 → iatoolkit-2.15.4}/src/iatoolkit/repositories/database_manager.py

@@ -18,7 +18,7 @@ import logging
 class DatabaseManager(DatabaseProvider):
     _POSTGRES_BOOTSTRAP_PATCHES = (
     )
-    _DEFAULT_CONNECT_TIMEOUT = 10
+    _DEFAULT_CONNECT_TIMEOUT = 60
 
     @inject
     def __init__(self,

{iatoolkit-2.15.2 → iatoolkit-2.15.4}/src/iatoolkit/repositories/document_repo.py

@@ -9,6 +9,7 @@ from injector import inject
 from iatoolkit.repositories.database_manager import DatabaseManager
 from iatoolkit.common.exceptions import IAToolkitException
 from typing import List, Optional
+from sqlalchemy import func
 
 
 class DocumentRepo:
@@ -54,9 +55,10 @@ class DocumentRepo:
         if not collection_name:
             return None
 
+        normalized_name = collection_name.strip().lower()
         ct = self.session.query(CollectionType).join(Company).filter(
             Company.short_name == company_short_name,
-            CollectionType.name == collection_name.strip().lower()
+            func.lower(CollectionType.name) == normalized_name
         ).first()
         return ct.id if ct else None
 
@@ -77,19 +79,23 @@ class DocumentRepo:
 
         collections = self.session.query(CollectionType).join(Company).filter(
             Company.short_name == company_short_name,
-            CollectionType.name.in_(normalized_names)
+            func.lower(CollectionType.name).in_(normalized_names)
         ).all()
 
-        collection_ids_by_name = {collection.name: collection.id for collection in collections}
+        collection_ids_by_name = {
+            str(collection.name or "").strip().lower(): collection.id
+            for collection in collections
+        }
         return [collection_ids_by_name[name] for name in normalized_names if name in collection_ids_by_name]
 
     def get_collection_by_name(self, company_short_name: str, collection_name: str) -> Optional[CollectionType]:
         if not collection_name:
             return None
 
+        normalized_name = collection_name.strip().lower()
         return self.session.query(CollectionType).join(Company).filter(
             Company.short_name == company_short_name,
-            CollectionType.name == collection_name.strip().lower()
+            func.lower(CollectionType.name) == normalized_name
         ).first()
 
     def get_collection_by_id(self, collection_id) -> Optional[CollectionType]:

{iatoolkit-2.15.2 → iatoolkit-2.15.4}/src/iatoolkit/repositories/llm_query_repo.py

@@ -36,10 +36,15 @@ class LLMQueryRepo:
         return query
 
     # get user query history
-    def get_history(self, company: Company, user_identifier: str) -> list[LLMQuery]:
-        return self.session.query(LLMQuery).filter(
-            LLMQuery.user_identifier == user_identifier,
-        ).filter_by(company_id=company.id).order_by(LLMQuery.created_at.desc()).limit(100).all()
+    def get_history(self, company: Company, user_identifier: str, limit: int = 100) -> list[LLMQuery]:
+        return (
+            self.session.query(LLMQuery)
+            .filter(LLMQuery.user_identifier == user_identifier)
+            .filter_by(company_id=company.id)
+            .order_by(LLMQuery.created_at.desc())
+            .limit(limit)
+            .all()
+        )
 
 
     ## --- Tools related methods

{iatoolkit-2.15.2 → iatoolkit-2.15.4}/src/iatoolkit/services/history_manager_service.py

@@ -193,14 +193,14 @@ class HistoryManagerService:
                 break
 
     # --- this is for the history popup in the chat page
-    def get_full_history(self, company_short_name: str, user_identifier: str) -> dict:
+    def get_full_history(self, company_short_name: str, user_identifier: str, limit: int = 100) -> dict:
         """Retrieves the full persisted history from the database."""
         try:
             company = self.profile_repo.get_company_by_short_name(company_short_name)
             if not company:
                 return {"error": self.i18n.t('errors.company_not_found', company_short_name=company_short_name)}
 
-            history = self.llm_query_repo.get_history(company, user_identifier)
+            history = self.llm_query_repo.get_history(company, user_identifier, limit=limit)
             if not history:
                 return {'message': 'empty history', 'history': []}
 

{iatoolkit-2.15.2 → iatoolkit-2.15.4}/src/iatoolkit/services/knowledge_base_service.py

@@ -578,7 +578,10 @@ class KnowledgeBaseService:
                 query = query.filter(Document.status == status)
 
         if collection:
-            query = query.join(Document.collection_type).filter(CollectionType.name == collection.lower())
+            normalized_collection = str(collection).strip().lower()
+            query = query.join(Document.collection_type).filter(
+                func.lower(CollectionType.name) == normalized_collection
+            )
 
         if user_identifier:
             query = query.filter(Document.user_identifier.ilike(f"%{user_identifier}%"))

{iatoolkit-2.15.2 → iatoolkit-2.15.4}/src/iatoolkit/views/history_api_view.py

@@ -17,6 +17,7 @@ class HistoryApiView(MethodView):
     Handles requests from the web UI to fetch a user's query history.
     Authentication is based on the active Flask session.
     """
+    HISTORY_MODAL_LIMIT = 50
 
     @inject
     def __init__(self,
@@ -41,7 +42,8 @@ class HistoryApiView(MethodView):
             # The service's signature should now only expect user_identifier.
             response = self.history_service.get_full_history(
                 company_short_name=company_short_name,
-                user_identifier=user_identifier
+                user_identifier=user_identifier,
+                limit=self.HISTORY_MODAL_LIMIT,
             )
 
             if "error" in response:

{iatoolkit-2.15.2 → iatoolkit-2.15.4/src/iatoolkit.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 2.15.2
+Version: 2.15.4
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT