goose-py 0.9.10__tar.gz → 0.9.11__tar.gz

{goose_py-0.9.10 → goose_py-0.9.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: goose-py
-Version: 0.9.10
+Version: 0.9.11
 Summary: A tool for AI workflows based on human-computer collaboration and structured output.
 Author-email: Nash Taylor <nash@chelle.ai>, Joshua Cook <joshua@chelle.ai>, Michael Sankur <michael@chelle.ai>
 Requires-Python: >=3.12

{goose_py-0.9.10 → goose_py-0.9.11}/goose/_internal/types/agent.py

@@ -18,14 +18,14 @@ class Base64MediaContent(str):
 
 class AIModel(StrEnum):
     # vertex (production Google, requires GCP environment)
-    VERTEX_PRO = "vertex_ai/gemini-1.5-pro"
-    VERTEX_FLASH = "vertex_ai/gemini-1.5-flash"
-    VERTEX_FLASH_8B = "vertex_ai/gemini-1.5-flash-8b"
+    VERTEX_PRO = "vertex_ai/gemini-2.0-pro"
+    VERTEX_FLASH = "vertex_ai/gemini-2.0-flash"
+    VERTEX_FLASH_8B = "vertex_ai/gemini-2.0-flash-8b"
 
     # gemini (publicly available, no GCP environment required)
-    GEMINI_PRO = "gemini/gemini-1.5-pro"
-    GEMINI_FLASH = "gemini/gemini-1.5-flash"
-    GEMINI_FLASH_8B = "gemini/gemini-1.5-flash-8b"
+    GEMINI_PRO = "gemini/gemini-2.0-pro"
+    GEMINI_FLASH = "gemini/gemini-2.0-flash"
+    GEMINI_FLASH_8B = "gemini/gemini-2.0-flash-8b"
 
 
 class UserMediaContentType(StrEnum):

{goose_py-0.9.10 → goose_py-0.9.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "goose-py"
-version = "0.9.10"
+version = "0.9.11"
 description = "A tool for AI workflows based on human-computer collaboration and structured output."
 readme = "README.md"
 authors = [

goose_py-0.9.11/tests/test_flows/test_meal_plan.py

@@ -0,0 +1,135 @@
+"""
+This example demonstrates how to use Goose to plan a meal.
+
+Meal planning is a good use case because it demonstrates the need for tasks to depend on the outputs of other tasks.
+You have your initial flow input, which is your overall health goals, and that's the starting point for breakfast.
+But then lunch is a function of both your goals and your breakfast.
+And then dinner is a function of your goals, your breakfast, and your lunch.
+"""
+
+import pytest
+
+from goose import Agent, FlowArguments, Result, flow, task
+from goose.agent import AIModel, SystemMessage, TextMessagePart, UserMessage
+
+
+class HealthGoals(FlowArguments):
+    goals: list[str]
+
+
+class Meal(Result):
+    dish_name: str
+    ingredients: list[str]
+    reason: str
+
+
+@task
+async def plan_breakfast(*, goals: list[str], agent: Agent) -> Meal:
+    formatted_goals = "\n".join(f"- {goal}" for goal in goals)
+    suggested_meal = await agent(
+        messages=[
+            UserMessage(
+                parts=[
+                    TextMessagePart(
+                        text=f"What should I make for breakfast? Here are my health goals:\n{formatted_goals}"
+                    )
+                ]
+            ),
+        ],
+        system=SystemMessage(
+            parts=[
+                TextMessagePart(
+                    text=(
+                        "You are a helpful assistant that helps plan breakfasts based on health goals. "
+                        "Explain why you built the meal you did."
+                    )
+                )
+            ]
+        ),
+        response_model=Meal,
+        model=AIModel.GEMINI_FLASH,
+        task_name="breakfast",
+    )
+    return suggested_meal
+
+
+@task
+async def plan_lunch(*, goals: list[str], breakfast: Meal, agent: Agent) -> Meal:
+    formatted_goals = "\n".join(f"- {goal}" for goal in goals)
+    formatted_breakfast = f"Breakfast: {breakfast.dish_name} ({', '.join(breakfast.ingredients)})"
+    suggested_meal = await agent(
+        messages=[
+            UserMessage(
+                parts=[
+                    TextMessagePart(text=f"What should I make for lunch? Here are my health goals:\n{formatted_goals}"),
+                    TextMessagePart(text=f"Here was my breakfast:\n{formatted_breakfast}"),
+                ]
+            ),
+        ],
+        system=SystemMessage(
+            parts=[
+                TextMessagePart(
+                    text=(
+                        "You are a helpful assistant that helps plan lunches based on health goals. "
+                        "Consider the breakfast you had when planning your lunch. Explain why you built the meal you did."
+                    ),
+                )
+            ]
+        ),
+        response_model=Meal,
+        model=AIModel.GEMINI_FLASH,
+        task_name="lunch",
+    )
+    return suggested_meal
+
+
+@task
+async def plan_dinner(*, goals: list[str], lunch: Meal, breakfast: Meal, agent: Agent) -> Meal:
+    formatted_goals = "\n".join(f"- {goal}" for goal in goals)
+    formatted_breakfast = f"Breakfast: {breakfast.dish_name} ({', '.join(breakfast.ingredients)})"
+    formatted_lunch = f"Lunch: {lunch.dish_name} ({', '.join(lunch.ingredients)})"
+    suggested_meal = await agent(
+        messages=[
+            UserMessage(
+                parts=[
+                    TextMessagePart(
+                        text=f"What should I make for dinner? Here are my health goals:\n{formatted_goals}"
+                    ),
+                    TextMessagePart(text=f"Here was my breakfast:\n{formatted_breakfast}"),
+                    TextMessagePart(text=f"Here was my lunch:\n{formatted_lunch}"),
+                ]
+            ),
+        ],
+        system=SystemMessage(
+            parts=[
+                TextMessagePart(
+                    text=(
+                        "You are a helpful assistant that helps plan dinners based on health goals. "
+                        "Consider the breakfast and lunch you had when planning your dinner. "
+                        "Explain why you built the meal you did."
+                    ),
+                )
+            ]
+        ),
+        response_model=Meal,
+        model=AIModel.GEMINI_FLASH,
+        task_name="dinner",
+    )
+    return suggested_meal
+
+
+@flow
+async def meal_plan(*, flow_arguments: HealthGoals, agent: Agent) -> None:
+    breakfast = await plan_breakfast(goals=flow_arguments.goals, agent=agent)
+    lunch = await plan_lunch(goals=flow_arguments.goals, breakfast=breakfast, agent=agent)
+    await plan_dinner(goals=flow_arguments.goals, lunch=lunch, breakfast=breakfast, agent=agent)
+
+
+@pytest.mark.asyncio
+async def test_meal_plan_flow() -> None:
+    health_goals = HealthGoals(goals=["lose weight", "have more energy"])
+    async with meal_plan.start_run(run_id="my-meal-plan") as run:
+        await meal_plan.generate(health_goals)
+
+    dinner = run.get(task=plan_dinner).result
+    assert isinstance(dinner.reason, str)

{goose_py-0.9.10 → goose_py-0.9.11}/uv.lock

@@ -234,7 +234,7 @@ wheels = [
 
 [[package]]
 name = "goose-py"
-version = "0.9.10"
+version = "0.9.11"
 source = { editable = "." }
 dependencies = [
     { name = "jsonpath-ng" },