Merge pull request #67 from a-akimov/python-client-model-constant
Refactor the Python client to define the model only once
This commit is contained in:
commit
944a709203
1 changed files with 5 additions and 2 deletions
|
|
@ -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,
|
||||||
)
|
)
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue