Add more functionality to signature description parsing
This commit is contained in:
@@ -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"
|
||||||
|
|
||||||
|
|||||||
@@ -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.
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user