flock-core 0.2.10__tar.gz → 0.2.12__tar.gz

{flock_core-0.2.10 → flock_core-0.2.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flock-core
-Version: 0.2.10
+Version: 0.2.12
 Summary: Declarative LLM Orchestration at Scale
 Author-email: Andre Ratzenberger <andre.ratzenberger@whiteduck.de>
 License-File: LICENSE
@@ -39,16 +39,8 @@ Description-Content-Type: text/markdown
 <p align="center">
 <img src="docs/img/flock.png" width="600"><br>
 <img alt="Dynamic TOML Badge" src="https://img.shields.io/badge/dynamic/toml?url=https%3A%2F%2Fraw.githubusercontent.com%2Fwhiteducksoftware%2Fflock%2Frefs%2Fheads%2Fmaster%2Fpyproject.toml&query=%24.project.version&style=for-the-badge&logo=pypi&label=pip%20version">
-<img alt="X (formerly Twitter) Follow" src="https://img.shields.io/twitter/follow/whiteduck_gmbh?style=for-the-badge&logo=X">
-
-
-
-
-
-
-
-
-
+<a href="https://www.linkedin.com/company/whiteduck" target="_blank"><img alt="LinkedIn" src="https://img.shields.io/badge/linkedin-%230077B5.svg?style=for-the-badge&logo=linkedin&logoColor=white&label=whiteduck"></a>
+<a href="https://bsky.app/profile/whiteduck-gmbh.bsky.social" target="_blank"><img alt="Bluesky" src="https://img.shields.io/badge/bluesky-Follow-blue?style=for-the-badge&logo=bluesky&logoColor=%23fff&color=%23333&labelColor=%230285FF&label=whiteduck-gmbh"></a>
 
 ## Overview
 

{flock_core-0.2.10 → flock_core-0.2.12}/README.md

@@ -1,16 +1,8 @@
 <p align="center">
 <img src="docs/img/flock.png" width="600"><br>
 <img alt="Dynamic TOML Badge" src="https://img.shields.io/badge/dynamic/toml?url=https%3A%2F%2Fraw.githubusercontent.com%2Fwhiteducksoftware%2Fflock%2Frefs%2Fheads%2Fmaster%2Fpyproject.toml&query=%24.project.version&style=for-the-badge&logo=pypi&label=pip%20version">
-<img alt="X (formerly Twitter) Follow" src="https://img.shields.io/twitter/follow/whiteduck_gmbh?style=for-the-badge&logo=X">
-
-
-
-
-
-
-
-
-
+<a href="https://www.linkedin.com/company/whiteduck" target="_blank"><img alt="LinkedIn" src="https://img.shields.io/badge/linkedin-%230077B5.svg?style=for-the-badge&logo=linkedin&logoColor=white&label=whiteduck"></a>
+<a href="https://bsky.app/profile/whiteduck-gmbh.bsky.social" target="_blank"><img alt="Bluesky" src="https://img.shields.io/badge/bluesky-Follow-blue?style=for-the-badge&logo=bluesky&logoColor=%23fff&color=%23333&labelColor=%230285FF&label=whiteduck-gmbh"></a>
 
 ## Overview
 

{flock_core-0.2.10 → flock_core-0.2.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flock-core"
-version = "0.2.10"
+version = "0.2.12"
 description = "Declarative LLM Orchestration at Scale"
 readme = "README.md"
 authors = [

{flock_core-0.2.10 → flock_core-0.2.12}/src/flock/config.py

@@ -30,7 +30,7 @@ OTEL_SQL_DATABASE_NAME = config("OTEL_SQL_DATABASE", "flock_events.db")
 OTEL_FILE_NAME = config("OTEL_FILE_NAME", "flock_events.jsonl")
 OTEL_ENABLE_SQL: bool = config("OTEL_ENABLE_SQL", True) == "True"
 OTEL_ENABLE_FILE: bool = config("OTEL_ENABLE_FILE", True) == "True"
-OTEL_ENABLE_JAEGER: bool = config("OTEL_ENABLE_JAEGER", True) == "True"
+OTEL_ENABLE_JAEGER: bool = config("OTEL_ENABLE_JAEGER", False) == "True"
 
 
 TELEMETRY = TelemetryConfig(

{flock_core-0.2.10 → flock_core-0.2.12}/src/flock/core/flock.py

@@ -99,31 +99,34 @@ class Flock:
         Returns:
             FlockAgent: The registered agent instance.
         """
-        if not agent.model:
-            agent.model = self.model
-            logger.debug(
-                f"Using default model for agent {agent.name}", model=self.model
-            )
+        with tracer.start_as_current_span("add_agent") as span:
+            span.set_attribute("agent_name", agent.name)
+            if not agent.model:
+                agent.model = self.model
+                logger.debug(
+                    f"Using default model for agent {agent.name}",
+                    model=self.model,
+                )
 
-        if agent.name in self.agents:
-            logger.warning(
-                f"Agent {agent.name} already exists, returning existing instance"
+            if agent.name in self.agents:
+                logger.warning(
+                    f"Agent {agent.name} already exists, returning existing instance"
+                )
+                return self.agents[agent.name]
+            logger.info("Adding new agent")
+
+            self.agents[agent.name] = agent
+            self.registry.register_agent(agent)
+            self.context.add_agent_definition(
+                type(agent), agent.name, agent.to_dict()
             )
-            return self.agents[agent.name]
-        logger.info("Adding new agent")
-
-        self.agents[agent.name] = agent
-        self.registry.register_agent(agent)
-        self.context.add_agent_definition(
-            type(agent), agent.name, agent.to_dict()
-        )
-
-        if hasattr(agent, "tools") and agent.tools:
-            for tool in agent.tools:
-                self.registry.register_tool(tool.__name__, tool)
-                logger.debug("Registered tool", tool_name=tool.__name__)
-        logger.success("Agent added successfully")
-        return agent
+
+            if hasattr(agent, "tools") and agent.tools:
+                for tool in agent.tools:
+                    self.registry.register_tool(tool.__name__, tool)
+                    logger.debug("Registered tool", tool_name=tool.__name__)
+            logger.success("Agent added successfully")
+            return agent
 
     def add_tool(self, tool_name: str, tool: callable):
         """Register a tool with the Flock system.
@@ -132,9 +135,12 @@ class Flock:
             tool_name (str): The name under which the tool will be registered.
             tool (callable): The tool function to register.
         """
-        logger.info("Registering tool", tool_name=tool_name)
-        self.registry.register_tool(tool_name, tool)
-        logger.debug("Tool registered successfully")
+        with tracer.start_as_current_span("add_tool") as span:
+            span.set_attribute("tool_name", tool_name)
+            span.set_attribute("tool", tool.__name__)
+            logger.info("Registering tool", tool_name=tool_name)
+            self.registry.register_tool(tool_name, tool)
+            logger.debug("Tool registered successfully")
 
     async def run_async(
         self,

{flock_core-0.2.10 → flock_core-0.2.12}/src/flock/core/flock_agent.py

@@ -14,6 +14,8 @@ from flock.core.mixin.dspy_integration import AgentType, DSPyIntegrationMixin
 from flock.core.mixin.prompt_parser import PromptParserMixin
 
 logger = get_logger("flock")
+
+
 from opentelemetry import trace
 
 tracer = trace.get_tracer(__name__)
@@ -157,7 +159,7 @@ class FlockAgent(BaseModel, ABC, PromptParserMixin, DSPyIntegrationMixin):
     )
 
     use_cache: bool = Field(
-        default=False,
+        default=True,
         description="Set to True to enable caching of the agent's results.",
     )
 

{flock_core-0.2.10 → flock_core-0.2.12}/src/flock/core/logging/telemetry.py

@@ -61,6 +61,7 @@ class TelemetryConfig:
         self.enable_jaeger = enable_jaeger
         self.enable_file = enable_file
         self.enable_sql = enable_sql
+        self.global_tracer = None
 
     def setup_tracing(self):
         """Set up OpenTelemetry tracing with the specified exporters."""
@@ -117,7 +118,7 @@ class TelemetryConfig:
         provider.add_span_processor(
             BaggageAttributeSpanProcessor(baggage_keys=["session_id", "run_id"])
         )
-
+        self.global_tracer = trace.get_tracer("flock")
         sys.excepthook = self.log_exception_to_otel
 
     def log_exception_to_otel(self, exc_type, exc_value, exc_traceback):

{flock_core-0.2.10 → flock_core-0.2.12}/src/flock/core/mixin/dspy_integration.py

@@ -79,20 +79,28 @@ class DSPyIntegrationMixin:
                 try:
                     field_type = eval(type_str, sys.modules[__name__].__dict__)
                 except Exception as e:
-                    print("Failed to evaluate type_str in __name__" + e)
+                    logger.warning(
+                        "Failed to evaluate type_str in __name__" + str(e)
+                    )
                     field_type = eval(
                         type_str, sys.modules["__main__"].__dict__
                     )
 
-            except Exception:
+            except Exception as ex:
                 # AREPL fix - var
+                logger.warning(
+                    "Failed to evaluate type_str in __main__" + str(ex)
+                )
                 try:
                     field_type = eval(
                         f"exec_locals.get('{type_str}')",
                         sys.modules["__main__"].__dict__,
                     )
-                except Exception as e:
-                    print(e)
+                except Exception as ex_arepl:
+                    logger.warning(
+                        "Failed to evaluate type_str in exec_locals"
+                        + str(ex_arepl)
+                    )
                     field_type = str
 
             return name, field_type, desc

{flock_core-0.2.10 → flock_core-0.2.12}/uv.lock

@@ -957,7 +957,7 @@ wheels = [
 
 [[package]]
 name = "flock-core"
-version = "0.2.10"
+version = "0.2.12"
 source = { editable = "." }
 dependencies = [
     { name = "cloudpickle" },

{flock_core-0.2.10 → flock_core-0.2.12}/src/flock/core/registry/agent_registry.py

@@ -2,12 +2,12 @@
 
 from collections.abc import Callable
 
-from opentelemetry import trace
-
 from flock.core.flock_agent import FlockAgent
 from flock.core.logging.logging import get_logger
 
 logger = get_logger("registry")
+from opentelemetry import trace
+
 tracer = trace.get_tracer(__name__)