Refactor the python client to define the model only once

This commit is contained in:
Alex Akimov 2025-11-09 18:12:00 +01:00
commit dc60b2a7f7

View file

@ -10,6 +10,9 @@ from dotenv import load_dotenv
load_dotenv() # load environment variables from .env load_dotenv() # load environment variables from .env
# Claude model constant
ANTHROPIC_MODEL = "claude-sonnet-4-5"
class MCPClient: class MCPClient:
def __init__(self): def __init__(self):
# Initialize session and client objects # Initialize session and client objects
@ -64,7 +67,7 @@ class MCPClient:
# Initial Claude API call # Initial Claude API call
response = self.anthropic.messages.create( response = self.anthropic.messages.create(
model="claude-sonnet-4-5", model=ANTHROPIC_MODEL,
max_tokens=1000, max_tokens=1000,
messages=messages, messages=messages,
tools=available_tools tools=available_tools
@ -97,7 +100,7 @@ class MCPClient:
# Get next response from Claude # Get next response from Claude
response = self.anthropic.messages.create( response = self.anthropic.messages.create(
model="claude-sonnet-4-5", model=ANTHROPIC_MODEL,
max_tokens=1000, max_tokens=1000,
messages=messages, messages=messages,
) )