diff --git a/README.md b/README.md index c850ea0..63c58c0 100644 --- a/README.md +++ b/README.md @@ -18,7 +18,7 @@ Interactive demo for [ignfab/geocontext](https://github.com/ignfab/geocontext#re | MODEL_NAME | The name of the model (see [LangGraph - create_react_agent](https://langchain-ai.github.io/langgraph/agents/models/#use-in-an-agent) / [init_chat_model](https://python.langchain.com/api_reference/langchain/chat_models/langchain.chat_models.base.init_chat_model.html)) | "anthropic:claude-sonnet-4-6" | | ANTHROPIC_API_KEY | Required from `anthropic:*` models (https://console.anthropic.com/settings/keys) | | | GOOGLE_API_KEY | Required from `google_genai:*` models (https://aistudio.google.com/api-keys) | | -| TEMPERATURE | Model temperature | 0 | +| TEMPERATURE | Model temperature | 0.0 | | DB_URI | URL of the PostgreSQL (`postgresql://postgres:ChangeIt@localhost:5432/geocontext`) or Redis (ex : `redis://default:ChangeIt@localhost:6379/0`) database. | None (use InMemorySaver) | | CONTACT_EMAIL | Email for the contact button. | "dev@localhost" | | GEOCONTEXT_LOG_LEVEL | Log level for Geocontext MCP. | error | diff --git a/agent.py b/agent.py index 0058e68..becee9a 100644 --- a/agent.py +++ b/agent.py @@ -11,17 +11,19 @@ from tools import create_map -# retreive model name from environment variable or use default +# retrieve model name from environment variable or use default MODEL_NAME = os.getenv("MODEL_NAME", "anthropic:claude-sonnet-4-6") # retrieve temperature from environment variable or use default -TEMPERATURE = int(os.getenv("TEMPERATURE",0)) +TEMPERATURE = float(os.getenv("TEMPERATURE", 0.0)) # ensure that the required environment variable is set for anthropic models if MODEL_NAME.startswith("anthropic:"): if os.getenv("ANTHROPIC_API_KEY", None) is None: raise ValueError("ANTHROPIC_API_KEY environment variable is required for anthropic models") + if TEMPERATURE < 0 or TEMPERATURE > 1: + raise ValueError("TEMPERATURE must be between 0 and 1") -logger.info(f"Create graph using model: {MODEL_NAME}") +logger.info(f"Create graph using model: {MODEL_NAME} with temperature: {TEMPERATURE}") model = init_chat_model(MODEL_NAME, temperature=TEMPERATURE) diff --git a/docker-compose.yaml b/docker-compose.yaml index fc8c5d5..d7e57ab 100644 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -7,7 +7,7 @@ services: - https_proxy environment: - MODEL_NAME=${MODEL_NAME:-anthropic:claude-sonnet-4-6} - - TEMPERATURE=${TEMPERATURE:-0} + - TEMPERATURE=${TEMPERATURE:-0.0} - ANTHROPIC_API_KEY - GOOGLE_API_KEY - HTTP_PROXY