feat: add Claude-Gemini collaboration and chat capabilities
- Add collaboration demo showing dynamic context requests - Implement chat tool for general conversations and brainstorming - Add tool selection guide with clear boundaries - Introduce models configuration system - Update prompts for better tool descriptions - Refactor server to remove redundant functionality - Add comprehensive tests for collaboration features - Enhance base tool with collaborative features This enables Claude to request additional context from Gemini during tool execution, improving analysis quality and accuracy. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
59
tools/models.py
Normal file
59
tools/models.py
Normal file
@@ -0,0 +1,59 @@
|
||||
"""
|
||||
Data models for tool responses and interactions
|
||||
"""
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
from typing import Literal, Optional, Dict, Any, List
|
||||
|
||||
|
||||
class ToolOutput(BaseModel):
|
||||
"""Standardized output format for all tools"""
|
||||
|
||||
status: Literal["success", "error", "requires_clarification"] = "success"
|
||||
content: str = Field(..., description="The main content/response from the tool")
|
||||
content_type: Literal["text", "markdown", "json"] = "text"
|
||||
metadata: Optional[Dict[str, Any]] = Field(default_factory=dict)
|
||||
|
||||
|
||||
class ClarificationRequest(BaseModel):
|
||||
"""Request for additional context or clarification"""
|
||||
|
||||
question: str = Field(..., description="Question to ask Claude for more context")
|
||||
files_needed: Optional[List[str]] = Field(
|
||||
default_factory=list, description="Specific files that are needed for analysis"
|
||||
)
|
||||
suggested_next_action: Optional[Dict[str, Any]] = Field(
|
||||
None,
|
||||
description="Suggested tool call with parameters after getting clarification",
|
||||
)
|
||||
|
||||
|
||||
class DiagnosticHypothesis(BaseModel):
|
||||
"""A debugging hypothesis with context and next steps"""
|
||||
|
||||
rank: int = Field(..., description="Ranking of this hypothesis (1 = most likely)")
|
||||
confidence: Literal["high", "medium", "low"] = Field(
|
||||
..., description="Confidence level"
|
||||
)
|
||||
hypothesis: str = Field(..., description="Description of the potential root cause")
|
||||
reasoning: str = Field(..., description="Why this hypothesis is plausible")
|
||||
next_step: str = Field(
|
||||
..., description="Suggested action to test/validate this hypothesis"
|
||||
)
|
||||
|
||||
|
||||
class StructuredDebugResponse(BaseModel):
|
||||
"""Enhanced debug response with multiple hypotheses"""
|
||||
|
||||
summary: str = Field(..., description="Brief summary of the issue")
|
||||
hypotheses: List[DiagnosticHypothesis] = Field(
|
||||
..., description="Ranked list of potential causes"
|
||||
)
|
||||
immediate_actions: List[str] = Field(
|
||||
default_factory=list,
|
||||
description="Immediate steps to take regardless of root cause",
|
||||
)
|
||||
additional_context_needed: Optional[List[str]] = Field(
|
||||
default_factory=list,
|
||||
description="Additional files or information that would help with analysis",
|
||||
)
|
||||
Reference in New Issue
Block a user