flock-core 0.2.9__tar.gz → 0.2.11__tar.gz

{flock_core-0.2.9 → flock_core-0.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flock-core
-Version: 0.2.9
+Version: 0.2.11
 Summary: Declarative LLM Orchestration at Scale
 Author-email: Andre Ratzenberger <andre.ratzenberger@whiteduck.de>
 License-File: LICENSE

{flock_core-0.2.9 → flock_core-0.2.11}/pyproject.toml

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

{flock_core-0.2.9 → flock_core-0.2.11}/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.9 → flock_core-0.2.11}/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.9 → flock_core-0.2.11}/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__)

{flock_core-0.2.9 → flock_core-0.2.11}/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.9 → flock_core-0.2.11}/src/flock/core/mixin/dspy_integration.py

@@ -69,20 +69,39 @@ class DSPyIntegrationMixin:
                 # TODO: We have to find a way to avoid using eval here.
                 # This is a security risk, as it allows arbitrary code execution.
                 # Figure out why the following code doesn't work as well as the eval.
+
+                # import dspy
+
                 # field_type = dspy.PythonInterpreter(
-                #     sys.modules[__name__].__dict__
-                #     | sys.modules["__main__"].__dict__
+                #     sys.modules[__name__].__dict__ | sys.modules["__main__"].__dict__
                 # ).execute(type_str)
 
                 try:
                     field_type = eval(type_str, sys.modules[__name__].__dict__)
-                except Exception:
+                except Exception as e:
+                    logger.warning(
+                        "Failed to evaluate type_str in __name__" + str(e)
+                    )
                     field_type = eval(
                         type_str, sys.modules["__main__"].__dict__
                     )
 
-            except Exception:
-                field_type = str
+            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 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.9 → flock_core-0.2.11}/src/flock/core/util/cli_helper.py

@@ -20,7 +20,7 @@ def display_banner():
 │ ▒█▀▀▀ █░░ █▀▀█ █▀▀ █░█ │
 │ ▒█▀▀▀ █░░ █░░█ █░░ █▀▄ │
 │ ▒█░░░ ▀▀▀ ▀▀▀▀ ▀▀▀ ▀░▀ │
-╰━━━━━━━━━v{__version__}━━━━━━━━━╯
+╰━━━━━━━━━v{__version__}━━━━━━━━╯
 🦆     🐤    🐧     🐓
 """,
         justify="center",

{flock_core-0.2.9 → flock_core-0.2.11}/uv.lock

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

{flock_core-0.2.9 → flock_core-0.2.11}/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__)