agentc-core 0.2.1a1__tar.gz

agentc_core-0.2.1a1/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.3
+Name: agentc-core
+Version: 0.2.1a1
+Summary: The core for the Couchbase Agent Catalog project.
+License: Apache-2.0
+Author: Glenn Galvizo
+Author-email: glenn.galvizo@couchbase.com
+Requires-Python: >=3.12
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: GitPython (>=3.1.43,<4.0.0)
+Requires-Dist: PyYAML (>=6.0.1,<7.0.0)
+Requires-Dist: couchbase (>=4.3.0,<5.0.0)
+Requires-Dist: datamodel-code-generator (>=0.26.0,<0.27.0)
+Requires-Dist: gitignore-parser (>=0.1.11,<0.2.0)
+Requires-Dist: jinja2 (>=3.1.4,<4.0.0)
+Requires-Dist: jsbeautifier (>=1.15.1,<2.0.0)
+Requires-Dist: jsonschema (>=4.23.0,<5.0.0)
+Requires-Dist: openapi-pydantic (>=0.5.1,<0.6.0)
+Requires-Dist: py-openapi-schema-to-json-schema (>=0.0.3,<0.0.4)
+Requires-Dist: pydantic (>=2.10.6,<3.0.0)
+Requires-Dist: requests (>=2.32.3,<3.0.0)
+Requires-Dist: semantic_version (>=2.10.0,<3.0.0)
+Requires-Dist: sentence-transformers (>=3.4.1,<4.0.0)
+Project-URL: Documentation, https://couchbaselabs.github.io/agent-catalog/api.html
+Project-URL: Repository, https://github.com/couchbaselabs/agent-catalog

agentc_core-0.2.1a1/agentc_core/__init__.py

@@ -0,0 +1,9 @@
+import agentc_core.catalog as catalog
+import agentc_core.config as config
+import agentc_core.provider as provider
+import agentc_core.tool as tool
+
+__all__ = ["tool", "provider", "catalog", "config"]
+
+# DO NOT edit this value, the plugin "poetry-dynamic-versioning" will automatically set this.
+__version__ = "0.0.0"

agentc_core-0.2.1a1/agentc_core/activity/__init__.py

@@ -0,0 +1,7 @@
+from .logger import BaseLogger
+from .logger import DBLogger
+from .logger import LocalLogger
+from .span import GlobalSpan
+from .span import Span
+
+__all__ = ["LocalLogger", "DBLogger", "BaseLogger", "Span", "GlobalSpan"]

agentc_core-0.2.1a1/agentc_core/activity/logger/__init__.py

@@ -0,0 +1,6 @@
+from .base import BaseLogger
+from .chain import ChainLogger
+from .db import DBLogger
+from .local import LocalLogger
+
+__all__ = ["LocalLogger", "DBLogger", "BaseLogger", "ChainLogger"]

agentc_core-0.2.1a1/agentc_core/activity/logger/base.py

@@ -0,0 +1,59 @@
+import abc
+import datetime
+import logging
+import uuid
+
+from ...version import VersionDescriptor
+from ..models.content import Content
+from ..models.log import Log
+
+logger = logging.getLogger(__name__)
+
+
+class BaseLogger(abc.ABC):
+    def __init__(self, catalog_version: VersionDescriptor, **kwargs):
+        self.catalog_version = catalog_version
+        self.annotations = kwargs
+
+    def log(
+        self,
+        content: Content,
+        span_name: list[str],
+        session_id: str,
+        log_id: str = None,
+        timestamp: datetime.datetime = None,
+        **kwargs,
+    ) -> Log:
+        # If the timestamp is not given, generate this value ourselves.
+        if timestamp is None:
+            timestamp = datetime.datetime.now().astimezone()
+
+        # Note: The accept call annotations take precedence over init-time annotations.
+        if len(self.annotations) > 0 and len(kwargs) > 0:
+            annotations = {**self.annotations, **kwargs}
+        elif len(self.annotations) > 0:
+            annotations = self.annotations
+        elif len(kwargs) > 0:
+            annotations = kwargs
+        else:
+            annotations = None
+
+        message = Log(
+            identifier=log_id or uuid.uuid4().hex,
+            timestamp=timestamp.isoformat(),
+            span=Log.Span(name=span_name, session=session_id),
+            content=content,
+            catalog_version=self.catalog_version,
+            annotations=annotations,
+        )
+        self._accept(message, message.model_dump(exclude_none=True, mode="json"))
+
+        # For debug, we'll pretty-print what we log.
+        if logger.isEnabledFor(logging.DEBUG):
+            logger.debug(f"Logging message: {message.model_dump_json(indent=2, exclude_none=True)}")
+
+        return message
+
+    @abc.abstractmethod
+    def _accept(self, log_obj: Log, log_json: dict):
+        pass

agentc_core-0.2.1a1/agentc_core/activity/logger/chain.py

@@ -0,0 +1,22 @@
+import logging.handlers
+
+from ..models.log import Log
+from .base import BaseLogger
+from .db import DBLogger
+from .local import LocalLogger
+
+logger = logging.getLogger(__name__)
+
+
+class ChainLogger(BaseLogger):
+    # TODO (GLENN): Add rollover to our Config class.
+    def __init__(self, local_logger: LocalLogger, db_logger: DBLogger, **kwargs):
+        if db_logger.catalog_version != local_logger.catalog_version:
+            raise ValueError("Catalog versions must match between remote and local-FS loggers!")
+        super(ChainLogger, self).__init__(catalog_version=local_logger.catalog_version, **kwargs)
+        self.db_logger = db_logger
+        self.local_logger = local_logger
+
+    def _accept(self, log_obj: Log, log_json: dict):
+        self.db_logger._accept(log_obj, log_json)
+        self.local_logger._accept(log_obj, log_json)

agentc_core-0.2.1a1/agentc_core/activity/logger/db.py

@@ -0,0 +1,43 @@
+import logging
+import textwrap
+
+from ...defaults import DEFAULT_ACTIVITY_LOG_COLLECTION
+from ...defaults import DEFAULT_ACTIVITY_SCOPE
+from .base import BaseLogger
+from agentc_core.activity.models.log import Log
+from agentc_core.config import RemoteCatalogConfig
+from agentc_core.remote.util.ddl import check_if_scope_collection_exist
+from agentc_core.version import VersionDescriptor
+
+logger = logging.getLogger(__name__)
+
+
+class DBLogger(BaseLogger):
+    def __init__(self, cfg: RemoteCatalogConfig, catalog_version: VersionDescriptor, **kwargs):
+        super().__init__(catalog_version=catalog_version, **kwargs)
+
+        # Get bucket ref
+        self.cluster = cfg.Cluster()
+        cb = self.cluster.bucket(cfg.bucket)
+
+        # Get the bucket manager
+        bucket_manager = cb.collections()
+
+        scope_collection_exist = check_if_scope_collection_exist(
+            bucket_manager, DEFAULT_ACTIVITY_SCOPE, DEFAULT_ACTIVITY_LOG_COLLECTION, False
+        )
+        if not scope_collection_exist:
+            raise ValueError(
+                textwrap.dedent(f"""
+                The collection {cfg.bucket}.{DEFAULT_ACTIVITY_SCOPE}.{DEFAULT_ACTIVITY_LOG_COLLECTION} does not exist.\n
+                Please use the 'agentc init' command to create this collection.\n
+                Execute 'agentc init --help' for more information.
+            """)
+            )
+
+        # get collection ref
+        cb_coll = cb.scope(DEFAULT_ACTIVITY_SCOPE).collection(DEFAULT_ACTIVITY_LOG_COLLECTION)
+        self.cb_coll = cb_coll
+
+    def _accept(self, log_obj: Log, log_json: dict):
+        self.cb_coll.insert(log_obj.identifier, log_json)

agentc_core-0.2.1a1/agentc_core/activity/logger/local.py

@@ -0,0 +1,47 @@
+import agentc_core.defaults
+import gzip
+import json
+import logging
+import logging.handlers
+import os
+import shutil
+
+from ...config import LocalCatalogConfig
+from .base import BaseLogger
+from agentc_core.activity.models.log import Log
+from agentc_core.version import VersionDescriptor
+
+logger = logging.getLogger(__name__)
+
+
+class LocalLogger(BaseLogger):
+    # TODO (GLENN): Add rollover to our Config class.
+    def __init__(
+        self, cfg: LocalCatalogConfig, catalog_version: VersionDescriptor, rollover: int = 128_000_000, **kwargs
+    ):
+        """
+        :param output: Output file to write the audit logs to.
+        :param catalog_version: Catalog version associated with this audit instance.
+        :param rollover: Maximum size (bytes) of a log-file before rollover. Set this field to 0 to never rollover.
+        """
+        super(LocalLogger, self).__init__(catalog_version=catalog_version, **kwargs)
+        self.audit_logger = logging.getLogger("AGENT_CATALOG_" + LocalLogger.__name__.upper())
+        self.audit_logger.setLevel(logging.INFO)
+        self.audit_logger.handlers.clear()
+        self.audit_logger.propagate = False
+
+        def compress_and_remove(source_log_file: str, dest_log_file: str):
+            with open(source_log_file, "rb") as input_fp, gzip.open(dest_log_file, "wb") as output_fp:
+                shutil.copyfileobj(input_fp, output_fp)
+            os.remove(source_log_file)
+
+        # We'll rotate log files and subsequently compress them when they get too large.
+        filename = cfg.ActivityPath() / agentc_core.defaults.DEFAULT_ACTIVITY_FILE
+        self.rotating_handler = logging.handlers.RotatingFileHandler(filename, maxBytes=rollover)
+        self.rotating_handler.rotator = compress_and_remove
+        self.rotating_handler.namer = lambda name: name + ".gz"
+        self.rotating_handler.setFormatter(logging.Formatter("%(message)s"))
+        self.audit_logger.addHandler(self.rotating_handler)
+
+    def _accept(self, log_obj: Log, log_json: dict):
+        self.audit_logger.info(json.dumps(log_json))