"""Anthropic API client for getting Claude responses."""
import os
from anthropic import Anthropic
from anthropic.types import TextBlock
from models import ContextEntry
class ClaudeClient:
"""Client for interacting with Anthropic's Claude API."""
def __init__(self, api_key: str | None = None, model: str | None = None) -> None:
"""Initialize the Claude client."""
self.api_key = api_key or os.getenv("ANTHROPIC_API_KEY")
if not self.api_key:
msg = "Anthropic API key must be provided or set in ANTHROPIC_API_KEY environment variable"
raise ValueError(msg)
self.model: str = model or os.getenv("MCP_TOOLZ_CLAUDE_MODEL") or "claude-sonnet-4-5-20250929"
self.client = Anthropic(api_key=self.api_key)
def get_second_opinion(self, context: ContextEntry, question: str | None = None) -> str:
"""Get Claude's second opinion on a context, or answer a specific question.
Args:
context: The context entry to analyze
question: Optional specific question to ask. If None, provides general second opinion.
"""
if question:
# Custom question mode
system_prompt = """You are a senior software engineering consultant answering questions about code, \
architecture decisions, and implementation plans.
Provide clear, actionable answers based on the context provided."""
user_content = self._format_context_for_claude(context, question)
else:
# Generic second opinion mode
system_prompt = """You are a senior software engineering consultant providing second opinions on code, \
architecture decisions, and implementation plans.
Your role is to:
- Provide constructive, balanced feedback
- Highlight both strengths and potential issues
- Suggest alternatives when appropriate
- Point out edge cases or security concerns
- Be concise but thorough
Format your response clearly with sections as needed."""
user_content = self._format_context_for_claude(context)
response = self.client.messages.create(
model=self.model,
max_tokens=4096,
system=system_prompt,
messages=[
{"role": "user", "content": user_content},
],
temperature=0.7,
)
if response.content and isinstance(response.content[0], TextBlock):
return response.content[0].text
return ""
def _format_context_for_claude(self, context: ContextEntry, question: str | None = None) -> str:
"""Format a context entry for Claude consumption.
Args:
context: The context entry to format
question: Optional specific question to append
"""
parts = [
f"# Context: {context.title}",
f"\n**Type:** {context.type}",
f"**Timestamp:** {context.timestamp.isoformat()}",
]
if context.tags:
parts.append(f"**Tags:** {', '.join(context.tags)}")
parts.append("\n## Content\n")
# Add specific content based on type
if context.content.messages:
parts.append("### Conversation\n")
for msg in context.content.messages:
parts.append(msg)
if context.content.code:
parts.append("### Code\n")
for file_path, code in context.content.code.items():
parts.append(f"**File:** `{file_path}`\n```\n{code}\n```\n")
if context.content.suggestions:
parts.append(f"### Suggestion\n{context.content.suggestions}\n")
if context.content.errors:
parts.append(f"### Error/Debug Info\n```\n{context.content.errors}\n```\n")
# Add question or default request
if question:
parts.append(f"\n---\n**Question:** {question}")
else:
parts.append("\n---\nPlease provide a second opinion on the above context.")
return "\n".join(parts)