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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: goose-py
-Version: 0.9.11
+Version: 0.9.12
 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.11 → goose_py-0.9.12}/goose/_internal/task.py

@@ -56,6 +56,9 @@ class Task[**P, R: Result]:
         flow_run = self.__get_current_flow_run()
         node_state = flow_run.get(task=self, index=index)
 
+        if len(node_state.conversation.result_messages) == 0:
+            raise Honk("Cannot refine a task that has not been initially generated")
+
         if context is not None:
             node_state.set_context(context=context)
         node_state.add_user_message(message=user_message)

{goose_py-0.9.11 → goose_py-0.9.12}/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-2.0-pro"
-    VERTEX_FLASH = "vertex_ai/gemini-2.0-flash"
-    VERTEX_FLASH_8B = "vertex_ai/gemini-2.0-flash-8b"
+    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"
 
     # gemini (publicly available, no GCP environment required)
-    GEMINI_PRO = "gemini/gemini-2.0-pro"
-    GEMINI_FLASH = "gemini/gemini-2.0-flash"
-    GEMINI_FLASH_8B = "gemini/gemini-2.0-flash-8b"
+    GEMINI_PRO = "gemini/gemini-1.5-pro"
+    GEMINI_FLASH = "gemini/gemini-1.5-flash"
+    GEMINI_FLASH_8B = "gemini/gemini-1.5-flash-8b"
 
 
 class UserMediaContentType(StrEnum):

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

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

{goose_py-0.9.11 → goose_py-0.9.12}/tests/test_refining.py

@@ -7,6 +7,7 @@ from pytest_mock import MockerFixture
 
 from goose import Agent, FlowArguments, Result, flow, task
 from goose.agent import SystemMessage, TextMessagePart, UserMessage
+from goose.errors import Honk
 
 
 class MyFlowArguments(FlowArguments):
@@ -67,3 +68,14 @@ async def test_refining() -> None:
     assert random_words[0].result.word == "__ADAPTED__"  # adapted
     assert random_words[1].result.word != "__ADAPTED__"  # not adapted
     assert random_words[2].result.word != "__ADAPTED__"  # not adapted
+
+
+@pytest.mark.asyncio
+@pytest.mark.usefixtures("generate_random_word_adapter")
+async def test_refining_before_generate_fails() -> None:
+    with pytest.raises(Honk):
+        async with sentence.start_run(run_id="2"):
+            await generate_random_word.refine(
+                user_message=UserMessage(parts=[TextMessagePart(text="Change it")]),
+                context=SystemMessage(parts=[TextMessagePart(text="Extra info")]),
+            )

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

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

goose_py-0.9.11/tests/test_flows/test_meal_plan.py

@@ -1,135 +0,0 @@
-"""
-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)