flock-core 0.3.8__tar.gz → 0.3.10__tar.gz

{flock_core-0.3.8 → flock_core-0.3.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flock-core
-Version: 0.3.8
+Version: 0.3.10
 Summary: Declarative LLM Orchestration at Scale
 Author-email: Andre Ratzenberger <andre.ratzenberger@whiteduck.de>
 License-File: LICENSE
@@ -30,17 +30,17 @@ Requires-Dist: python-box>=7.3.2
 Requires-Dist: python-decouple>=3.8
 Requires-Dist: questionary>=2.1.0
 Requires-Dist: rich>=13.9.4
-Requires-Dist: sentence-transformers>=3.4.1
 Requires-Dist: temporalio>=1.9.0
 Requires-Dist: tiktoken>=0.8.0
 Requires-Dist: toml>=0.10.2
 Requires-Dist: tqdm>=4.67.1
 Requires-Dist: uvicorn>=0.34.0
 Requires-Dist: zep-python>=2.0.2
-Provides-Extra: all-tools
-Requires-Dist: docling>=2.18.0; extra == 'all-tools'
-Requires-Dist: markdownify>=0.14.1; extra == 'all-tools'
-Requires-Dist: tavily-python>=0.5.0; extra == 'all-tools'
+Provides-Extra: all
+Requires-Dist: docling>=2.18.0; extra == 'all'
+Requires-Dist: markdownify>=0.14.1; extra == 'all'
+Requires-Dist: sentence-transformers>=3.4.1; extra == 'all'
+Requires-Dist: tavily-python>=0.5.0; extra == 'all'
 Provides-Extra: tools
 Requires-Dist: markdownify>=0.14.1; extra == 'tools'
 Requires-Dist: tavily-python>=0.5.0; extra == 'tools'

{flock_core-0.3.8 → flock_core-0.3.10}/examples/01_introduction/01_simple_example.py

@@ -1,3 +1,4 @@
+import os
 from flock.core import Flock, FlockFactory
 from flock.core.flock_api import FlockAPI
 
@@ -8,7 +9,9 @@ from flock.core.flock_api import FlockAPI
 # Flock uses litellm to talk to LLMs
 # Please consult the litellm documentation for valid IDs:
 # https://docs.litellm.ai/docs/providers
-MODEL = "openai/gpt-4o"
+
+
+MODEL = "azure/ara-gpt4o"
 
 
 # --------------------------------

{flock_core-0.3.8 → flock_core-0.3.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flock-core"
-version = "0.3.8"
+version = "0.3.10"
 description = "Declarative LLM Orchestration at Scale"
 readme = "README.md"
 authors = [
@@ -35,7 +35,6 @@ dependencies = [
     "python-decouple>=3.8",
     "questionary>=2.1.0",
     "rich>=13.9.4",
-    "sentence-transformers>=3.4.1",
     "temporalio>=1.9.0",
     "tiktoken>=0.8.0",
     "toml>=0.10.2",
@@ -45,13 +44,16 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-all-tools = ["docling>=2.18.0", "tavily-python>=0.5.0", "markdownify>=0.14.1"]
+all = [
+    "docling>=2.18.0",
+    "tavily-python>=0.5.0",
+    "markdownify>=0.14.1",
+    "sentence-transformers>=3.4.1",
+]
 tools = ["tavily-python>=0.5.0", "markdownify>=0.14.1"]
 
 
 [dependency-groups]
-app = ["monsterui>=0.0.23", "python-fasthtml>=0.12.0"]
-tools = ["tavily-python>=0.5.0", "markdownify>=0.14.1"]
 dev = [
     "pytest>=8.3.3",
     "pytest-clarity>=1.0.1",

{flock_core-0.3.8 → flock_core-0.3.10}/src/flock/core/context/context_manager.py

@@ -5,6 +5,7 @@ from flock.core.context.context_vars import (
     FLOCK_CURRENT_AGENT,
     FLOCK_INITIAL_INPUT,
     FLOCK_LOCAL_DEBUG,
+    FLOCK_MODEL,
     FLOCK_RUN_ID,
 )
 
@@ -15,6 +16,7 @@ def initialize_context(
     input_data: dict,
     run_id: str,
     local_debug: bool,
+    model: str,
 ) -> None:
     """Initialize the FlockContext with standard variables before running an agent.
 
@@ -32,3 +34,4 @@ def initialize_context(
     context.set_variable(FLOCK_LOCAL_DEBUG, local_debug)
     context.run_id = run_id
     context.set_variable(FLOCK_RUN_ID, run_id)
+    context.set_variable(FLOCK_MODEL, model)

{flock_core-0.3.8 → flock_core-0.3.10}/src/flock/core/context/context_vars.py

@@ -6,3 +6,4 @@ FLOCK_LOCAL_DEBUG = "flock.local_debug"
 FLOCK_RUN_ID = "flock.run_id"
 FLOCK_LAST_AGENT = "flock.last_agent"
 FLOCK_LAST_RESULT = "flock.last_result"
+FLOCK_MODEL = "flock.model"

{flock_core-0.3.8 → flock_core-0.3.10}/src/flock/core/flock.py

@@ -135,7 +135,7 @@ class Flock:
         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
+                agent.set_model(self.model)
                 logger.debug(
                     f"Using default model for agent {agent.name}",
                     model=self.model,
@@ -443,6 +443,7 @@ class Flock:
                     self.input,
                     run_id,
                     not self.enable_temporal,
+                    self.model,
                 )
 
                 logger.info(

{flock_core-0.3.8 → flock_core-0.3.10}/src/flock/core/flock_agent.py

@@ -200,6 +200,11 @@ class FlockAgent(BaseModel, ABC):
         """Run the agent with the given inputs and return its generated output."""
         return asyncio.run(self.run_async(inputs))
 
+    def set_model(self, model: str):
+        """Set the model for the agent."""
+        self.model = model
+        self.evaluator.config.model = model
+
     async def run_async(self, inputs: dict[str, Any]) -> dict[str, Any]:
         with tracer.start_as_current_span("agent.run") as span:
             span.set_attribute("agent.name", self.name)

{flock_core-0.3.8 → flock_core-0.3.10}/src/flock/core/flock_factory.py

@@ -23,7 +23,7 @@ class FlockFactory:
     def create_default_agent(
         name: str,
         description: str | Callable[..., str] | None = None,
-        model: str | Callable[..., str] | None = "openai/gpt-4o",
+        model: str | Callable[..., str] | None = None,
         input: str | Callable[..., str] | None = None,
         output: str | Callable[..., str] | None = None,
         tools: list[Callable[..., Any] | Any] | None = None,

{flock_core-0.3.8 → flock_core-0.3.10}/src/flock/core/util/cli_helper.py

@@ -52,7 +52,7 @@ def init_console(clear_screen: bool = True):
 │ ▒█▀▀▀ █░░ █▀▀█ █▀▀ █░█ │
 │ ▒█▀▀▀ █░░ █░░█ █░░ █▀▄ │
 │ ▒█░░░ ▀▀▀ ▀▀▀▀ ▀▀▀ ▀░▀ │
-╰━━━━━━━━━v{__version__}━━━━━━━━━╯
+╰━━━━━━━━━v{__version__}━━━━━━━━╯
 🦆     🐤    🐧     🐓
 """,
         justify="center",
@@ -83,5 +83,3 @@ def display_banner_no_version():
     )
     console.print(banner_text)
     console.print(f"[bold]white duck GmbH[/] - [cyan]https://whiteduck.de[/]\n")
-
-