pixie-examples 0.1.1.dev21__tar.gz → 0.2.3__tar.gz

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pixie-examples
-Version: 0.1.1.dev21
+Version: 0.2.3
 Summary: examples for using Pixie
 License: MIT
 License-File: LICENSE
@@ -24,7 +24,7 @@ Requires-Dist: openinference-instrumentation-crewai (>=0.1.17,<0.2.0)
 Requires-Dist: openinference-instrumentation-dspy (>=0.1.33,<0.2.0)
 Requires-Dist: openinference-instrumentation-google-adk (>=0.1.8,<0.2.0)
 Requires-Dist: openinference-instrumentation-openai-agents (>=1.4.0,<2.0.0)
-Requires-Dist: pixie-sdk (>=0.1.1.dev42,<0.2.0)
+Requires-Dist: pixie-sdk (==0.2.3)
 Requires-Dist: pydantic (>=2.7.4,<3.0.0)
 Requires-Dist: pydantic-ai-slim (>=1.39.0,<2.0.0)
 Requires-Dist: pymarkdownlnt (>=0.9.34,<0.10.0)

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/langchain/basic_agent.py

@@ -9,7 +9,7 @@ from langchain.agents import create_agent
 from langchain.chat_models import init_chat_model
 
 from langfuse.langchain import CallbackHandler
-import pixie
+import pixie.sdk as pixie
 
 
 basic_weather_agent_prompt = pixie.create_prompt(

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/langchain/customer_support.py

@@ -20,7 +20,7 @@ from langfuse.langchain import CallbackHandler
 from langchain_core.messages import (
     SystemMessage,
 )
-import pixie
+import pixie.sdk as pixie
 
 
 langfuse_handler = CallbackHandler()
@@ -93,11 +93,11 @@ def provide_solution(solution: str) -> str:
     return f"Solution provided: {solution}"
 
 
-class IssueClassifierPromptVariables(pixie.PromptVariables):
+class IssueClassifierVariables(pixie.Variables):
     warranty_status: Literal["in_warranty", "out_of_warranty"]
 
 
-class ResolutionSpecialistPromptVariables(IssueClassifierPromptVariables):
+class ResolutionSpecialistVariables(IssueClassifierVariables):
     issue_type: Literal["hardware", "software"]
 
 
@@ -107,12 +107,12 @@ warranty_collector_prompt = pixie.create_prompt(
 )
 issue_classifier_prompt = pixie.create_prompt(
     "issue_classifier_agent",
-    IssueClassifierPromptVariables,
+    IssueClassifierVariables,
     description="Customer support agent that classifies technical issues as hardware or software",
 )
 resolution_specialist_prompt = pixie.create_prompt(
     "resolution_specialist_agent",
-    ResolutionSpecialistPromptVariables,
+    ResolutionSpecialistVariables,
     description="Customer support agent that provides resolutions based on issue type and warranty status",
 )
 # Step configuration

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/langchain/personal_assistant.py

@@ -13,7 +13,7 @@ from langchain.tools import tool
 from langgraph.checkpoint.memory import InMemorySaver
 
 from langfuse.langchain import CallbackHandler
-import pixie
+import pixie.sdk as pixie
 
 
 calendar_agent_prompt = pixie.create_prompt(

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/langchain/sql_agent.py

@@ -22,18 +22,18 @@ from langchain.chat_models import init_chat_model
 from langgraph.checkpoint.memory import InMemorySaver
 
 from langfuse.langchain import CallbackHandler
-import pixie
+import pixie.sdk as pixie
 from ..sql_utils import SQLDatabase, SQLDatabaseToolkit
 
 
-class SqlAgentPromptVariables(pixie.PromptVariables):
+class SqlAgentVariables(pixie.Variables):
     dialect: str
     top_k: int
 
 
 sql_agent_prompt = pixie.create_prompt(
     "langchain_sql_agent",
-    SqlAgentPromptVariables,
+    SqlAgentVariables,
     description="SQL agent that interacts with databases and creates queries based on natural language",
 )
 
@@ -86,7 +86,7 @@ async def langchain_sql_query_agent(question: str) -> str:
 
     # Format system prompt with database info
     system_prompt = sql_agent_prompt.compile(
-        SqlAgentPromptVariables(dialect=db.dialect, top_k=5)
+        SqlAgentVariables(dialect=db.dialect, top_k=5)
     )
 
     # Create agent
@@ -123,7 +123,7 @@ async def langchain_interactive_sql_agent() -> pixie.PixieGenerator[str, str]:
 
     # Format system prompt
     system_prompt = sql_agent_prompt.compile(
-        SqlAgentPromptVariables(dialect=db.dialect, top_k=5)
+        SqlAgentVariables(dialect=db.dialect, top_k=5)
     )
 
     # Create agent with checkpointer for conversation memory

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/langgraph/langgraph_rag.py

@@ -23,38 +23,38 @@ from langgraph.graph import END, START, MessagesState, StateGraph
 from langgraph.prebuilt import ToolNode, tools_condition
 
 from langfuse.langchain import CallbackHandler
-import pixie
+import pixie.sdk as pixie
 import requests
 from bs4 import BeautifulSoup
 
 
-class GradePromptVariables(pixie.PromptVariables):
+class GradeVariables(pixie.Variables):
     context: str
     question: str
 
 
-class RewritePromptVariables(pixie.PromptVariables):
+class RewriteVariables(pixie.Variables):
     question: str
 
 
-class GeneratePromptVariables(pixie.PromptVariables):
+class GenerateVariables(pixie.Variables):
     question: str
     context: str
 
 
 rag_grade_prompt = pixie.create_prompt(
     "rag_grade_documents",
-    GradePromptVariables,
+    GradeVariables,
     description="Grades relevance of retrieved documents to user questions",
 )
 rag_rewrite_prompt = pixie.create_prompt(
     "rag_rewrite_question",
-    RewritePromptVariables,
+    RewriteVariables,
     description="Rewrites questions to improve semantic understanding",
 )
 rag_generate_prompt = pixie.create_prompt(
     "rag_generate_answer",
-    GeneratePromptVariables,
+    GenerateVariables,
     description="Generates concise answers from retrieved context",
 )
 
@@ -143,7 +143,7 @@ def create_rag_graph(retriever, model):
         context = cast(str, state["messages"][-1].content)
 
         prompt = rag_grade_prompt.compile(
-            GradePromptVariables(question=question, context=context)
+            GradeVariables(question=question, context=context)
         )
         response = grader_model.with_structured_output(GradeDocuments).invoke(
             [{"role": "user", "content": prompt}],
@@ -161,7 +161,7 @@ def create_rag_graph(retriever, model):
         """Rewrite the original user question."""
         messages = state["messages"]
         question = cast(str, messages[0].content)
-        prompt = rag_rewrite_prompt.compile(RewritePromptVariables(question=question))
+        prompt = rag_rewrite_prompt.compile(RewriteVariables(question=question))
         response = model.invoke(
             [{"role": "user", "content": prompt}],
             config={"callbacks": [langfuse_handler]},
@@ -174,7 +174,7 @@ def create_rag_graph(retriever, model):
         question = cast(str, state["messages"][0].content)
         context = cast(str, state["messages"][-1].content)
         prompt = rag_generate_prompt.compile(
-            GeneratePromptVariables(question=question, context=context)
+            GenerateVariables(question=question, context=context)
         )
         response = model.invoke(
             [{"role": "user", "content": prompt}],

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/langgraph/langgraph_sql_agent.py

@@ -19,21 +19,21 @@ from langgraph.prebuilt import ToolNode
 from langgraph.checkpoint.memory import InMemorySaver
 
 from langfuse.langchain import CallbackHandler
-import pixie
+import pixie.sdk as pixie
 
 
-class LanggraphSqlPromptVariables(pixie.PromptVariables):
+class LanggraphSqlVariables(pixie.Variables):
     dialect: str
 
 
 langgraph_sql_generate_prompt = pixie.create_prompt(
     "langgraph_sql_generate_query",
-    LanggraphSqlPromptVariables,
+    LanggraphSqlVariables,
     description="Generates SQL queries from natural language questions",
 )
 langgraph_sql_check_prompt = pixie.create_prompt(
     "langgraph_sql_check_query",
-    LanggraphSqlPromptVariables,
+    LanggraphSqlVariables,
     description="Reviews and validates SQL queries for common mistakes",
 )
 
@@ -101,7 +101,7 @@ def create_sql_graph(db: SQLDatabase, model):
     # Node: Generate query
     def generate_query(state: MessagesState):
         generate_query_prompt_text = langgraph_sql_generate_prompt.compile(
-            LanggraphSqlPromptVariables(dialect=db.dialect)
+            LanggraphSqlVariables(dialect=db.dialect)
         )
         system_message = {"role": "system", "content": generate_query_prompt_text}
         llm_with_tools = model.bind_tools([run_query_tool])
@@ -116,7 +116,7 @@ def create_sql_graph(db: SQLDatabase, model):
         from langchain.messages import AIMessage as AI
 
         check_query_prompt_text = langgraph_sql_check_prompt.compile(
-            LanggraphSqlPromptVariables(dialect=db.dialect)
+            LanggraphSqlVariables(dialect=db.dialect)
         )
         system_message = {"role": "system", "content": check_query_prompt_text}
         last_message = state["messages"][-1]

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/openai_agents_sdk/customer_service.py

@@ -26,7 +26,7 @@ from agents import (
     function_tool,
     handoff,
 )
-import pixie
+import pixie.sdk as pixie
 
 
 faq_agent_prompt = pixie.create_prompt(

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/openai_agents_sdk/financial_research_agent.py

@@ -22,7 +22,7 @@ from collections.abc import Sequence
 
 from pydantic import BaseModel
 from agents import Agent, Runner, RunResult, RunResultStreaming, WebSearchTool
-import pixie
+import pixie.sdk as pixie
 
 
 financial_planner_prompt = pixie.create_prompt(

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/openai_agents_sdk/llm_as_a_judge.py

@@ -14,7 +14,7 @@ from dataclasses import dataclass
 from typing import Literal
 
 from agents import Agent, ItemHelpers, Runner, TResponseInputItem
-import pixie
+import pixie.sdk as pixie
 
 
 @dataclass

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/openai_agents_sdk/routing.py

@@ -13,7 +13,7 @@ from __future__ import annotations
 
 from openai.types.responses import ResponseContentPartDoneEvent, ResponseTextDeltaEvent
 from agents import Agent, RawResponsesStreamEvent, Runner, TResponseInputItem
-import pixie
+import pixie.sdk as pixie
 
 
 french_agent_prompt = pixie.create_prompt(

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/pydantic_ai/README.md

@@ -231,7 +231,7 @@ subscription {
 All examples use:
 
 ```python
-from pixie import app, PixieGenerator, UserInputRequirement
+from pixie.sdk import app, PixieGenerator, UserInputRequirement
 
 @app
 async def my_handler(input: InputType) -> OutputType:

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/pydantic_ai/bank_support.py

@@ -23,7 +23,7 @@ import sqlite3
 from dataclasses import dataclass
 from pydantic import BaseModel
 from pydantic_ai import Agent, RunContext
-import pixie
+import pixie.sdk as pixie
 
 
 bank_support_agent_prompt = pixie.create_prompt(

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/pydantic_ai/flight_booking.py

@@ -16,7 +16,7 @@ from pydantic_ai import (
     RunUsage,
     UsageLimits,
 )
-import pixie
+import pixie.sdk as pixie
 
 
 flight_search_agent_prompt = pixie.create_prompt(

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/pydantic_ai/question_graph.py

@@ -13,7 +13,7 @@ from pydantic_graph import (
     Graph,
     GraphRunContext,
 )
-import pixie
+import pixie.sdk as pixie
 
 
 class EvaluationOutput(BaseModel, use_attribute_docstrings=True):

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/pydantic_ai/sql_gen.py

@@ -10,10 +10,10 @@ from typing import Annotated, TypeAlias
 from annotated_types import MinLen
 from pydantic import BaseModel, Field
 from pydantic_ai import Agent, ModelRetry, RunContext, format_as_xml
-import pixie
+import pixie.sdk as pixie
 
 
-class SqlGenPromptVariables(pixie.PromptVariables):
+class SqlGenVariables(pixie.Variables):
     db_schema: str
     today_date: str
     sql_examples: str
@@ -21,7 +21,7 @@ class SqlGenPromptVariables(pixie.PromptVariables):
 
 sql_gen_agent_prompt = pixie.create_prompt(
     "sql_gen_agent",
-    SqlGenPromptVariables,
+    SqlGenVariables,
     description="Generates SQL queries from natural language for PostgreSQL databases",
 )
 
@@ -102,7 +102,7 @@ agent = Agent[Deps, Response](
 @agent.system_prompt
 async def system_prompt() -> str:
     return sql_gen_agent_prompt.compile(
-        SqlGenPromptVariables(
+        SqlGenVariables(
             db_schema=DB_SCHEMA,
             today_date=str(date.today()),
             sql_examples=format_as_xml(SQL_EXAMPLES),

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/pydantic_ai/structured_output.py

@@ -20,7 +20,7 @@ Then query via GraphQL:
 import os
 from pydantic import BaseModel
 from pydantic_ai import Agent
-import pixie
+import pixie.sdk as pixie
 
 
 class MyModel(BaseModel):

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/quickstart/chatbot.py

@@ -1,6 +1,6 @@
 from pydantic_ai import Agent
 
-import pixie
+import pixie.sdk as pixie
 
 
 prompt = pixie.create_prompt(

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/quickstart/problem_solver.py

@@ -1,10 +1,10 @@
 from pydantic import Field
 from pydantic_ai import Agent
 
-import pixie
+import pixie.sdk as pixie
 
 
-class PromptVariables(pixie.PromptVariables):
+class Variables(pixie.Variables):
     problem_description: str = Field(
         ..., description="A description of the problem to be solved."
     )
@@ -12,7 +12,7 @@ class PromptVariables(pixie.PromptVariables):
 
 agent_prompt = pixie.create_prompt(
     "problem_solver",
-    PromptVariables,
+    Variables,
     description="Problem solving agent that thinks step by step",
 )
 
@@ -30,7 +30,7 @@ async def example_problem_solver(problem_description: str):
     agent = Agent(
         name=agent_prompt.id,
         instructions=agent_prompt.compile(
-            PromptVariables(problem_description=problem_description)
+            Variables(problem_description=problem_description)
         ),
         model="gpt-4o-mini",
     )

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/quickstart/sleepy_poet.py

@@ -7,7 +7,7 @@ from pydantic import BaseModel, Field
 from pydantic_ai import Agent, RunContext
 from pydantic_ai.models.openai import OpenAIChatModelSettings
 
-import pixie
+import pixie.sdk as pixie
 
 sleepy_poet_prompt = pixie.create_prompt(
     "sleepy_poet",

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/examples/quickstart/weather_agent.py

@@ -6,7 +6,7 @@ from typing import Any
 from httpx import AsyncClient
 from pydantic import BaseModel
 from pydantic_ai import Agent, RunContext
-import pixie
+import pixie.sdk as pixie
 
 
 @dataclass

{pixie_examples-0.1.1.dev21 → pixie_examples-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ packages = [
     { include = "examples" }
 ]
 
-version = "0.1.1.dev21"  # Managed by setuptools-scm
+version = "0.2.3"  # Managed by setuptools-scm
 description = "examples for using Pixie"
 authors = ["Yiou Li <yol@gopixie.ai>"]
 license = "MIT"
@@ -21,7 +21,7 @@ Changelog = "https://github.com/yiouli/pixie-examples/commits/main/"
 python = ">=3.11,<3.14"
 pydantic-ai-slim = "^1.39.0"
 pydantic = {version = "^2.7.4", extras = []}
-pixie-sdk = "^0.1.1.dev42"
+pixie-sdk = "0.2.3"
 openai-agents = "^0.6.5"
 pymarkdownlnt = "^0.9.34"
 langchain = "^1.2.3"