Add more functionality to signature description parsing

This commit is contained in:
2025-12-06 00:07:42 -05:00
parent e1c81644c3
commit b5284d344b
2 changed files with 9 additions and 6 deletions

View File

@@ -1,5 +1,5 @@
from claude_dspy.agent import ClaudeCode, ClaudeCodeConfig from .agent import ClaudeCode, ClaudeCodeConfig
from claude_dspy.trace import ( from .trace import (
TraceItem, TraceItem,
AgentMessageItem, AgentMessageItem,
ThinkingItem, ThinkingItem,
@@ -7,7 +7,7 @@ from claude_dspy.trace import (
ToolResultItem, ToolResultItem,
ErrorItem, ErrorItem,
) )
from claude_dspy.utils import Usage from .utils import Usage
__version__ = "0.1.0" __version__ = "0.1.0"

View File

@@ -22,7 +22,7 @@ from claude_agent_sdk import (
ToolResultBlock, ToolResultBlock,
) )
from claude_dspy.trace import ( from .trace import (
TraceItem, TraceItem,
AgentMessageItem, AgentMessageItem,
ThinkingItem, ThinkingItem,
@@ -30,7 +30,7 @@ from claude_dspy.trace import (
ToolResultItem, ToolResultItem,
ErrorItem, ErrorItem,
) )
from claude_dspy.utils import ( from .utils import (
Usage, Usage,
is_pydantic_model, is_pydantic_model,
get_json_schema, get_json_schema,
@@ -41,8 +41,10 @@ from claude_dspy.utils import (
class ClaudeCodeConfig(PrecompiledConfig): class ClaudeCodeConfig(PrecompiledConfig):
"""Configuration for ClaudeCode agent.""" """Configuration for ClaudeCode agent."""
model: str = "claude-opus-4-5-20251101" model: str = "claude-opus-4-5-20251101"
class ClaudeCodeKwargs(BaseModel): class ClaudeCodeKwargs(BaseModel):
model_config = {"arbitrary_types_allowed": True} model_config = {"arbitrary_types_allowed": True}
@@ -55,6 +57,7 @@ class ClaudeCodeKwargs(BaseModel):
sandbox: dict[str, Any] | None = None sandbox: dict[str, Any] | None = None
system_prompt: str | dict[str, Any] | None = None system_prompt: str | dict[str, Any] | None = None
class ClaudeCode(PrecompiledProgram): class ClaudeCode(PrecompiledProgram):
"""DSPy module that wraps Claude Code SDK. """DSPy module that wraps Claude Code SDK.