Spaces:
Running
Running
File size: 4,055 Bytes
676582c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 |
"""
Agent Configuration
Configuration dataclass for agent behavior and LLM provider settings.
"""
from dataclasses import dataclass
from typing import Optional
@dataclass
class AgentConfiguration:
"""
Configuration for the AI agent.
This dataclass defines all configurable parameters for agent behavior,
including LLM provider settings, conversation limits, and system prompts.
"""
# Provider settings
provider: str = "gemini" # Options: gemini, openrouter, cohere
fallback_provider: Optional[str] = None # Optional fallback provider
model: Optional[str] = None # Model name (provider-specific)
# API keys (loaded from environment)
gemini_api_key: Optional[str] = None
openrouter_api_key: Optional[str] = None
cohere_api_key: Optional[str] = None
# Generation parameters
temperature: float = 0.7 # Sampling temperature (0.0 to 1.0)
max_tokens: int = 8192 # Maximum tokens in response
# Conversation history limits (for free-tier constraints)
max_messages: int = 20 # Maximum messages to keep in history
max_conversation_tokens: int = 8000 # Maximum tokens in conversation history
# System prompt
system_prompt: str = """You are a helpful AI assistant for managing tasks.
You can help users create, view, complete, update, and delete tasks using natural language.
Available tools:
- add_task: Create a new task
- list_tasks: View all tasks (with optional filtering)
- complete_task: Mark a task as completed
- delete_task: Remove a task
- update_task: Modify task properties
Always respond in a friendly, conversational manner and confirm actions taken."""
# Retry settings
max_retries: int = 3 # Maximum retries on rate limit errors
retry_delay: float = 1.0 # Delay between retries (seconds)
def get_provider_api_key(self, provider_name: str) -> Optional[str]:
"""
Get API key for a specific provider.
Args:
provider_name: Provider name (gemini, openrouter, cohere)
Returns:
API key or None if not configured
"""
if provider_name == "gemini":
return self.gemini_api_key
elif provider_name == "openrouter":
return self.openrouter_api_key
elif provider_name == "cohere":
return self.cohere_api_key
return None
def get_provider_model(self, provider_name: str) -> str:
"""
Get default model for a specific provider.
Args:
provider_name: Provider name
Returns:
Model identifier
"""
if self.model:
return self.model
# Default models per provider
defaults = {
"gemini": "gemini-flash-latest",
"openrouter": "google/gemini-flash-1.5",
"cohere": "command-r-plus"
}
return defaults.get(provider_name, "gemini-flash-latest")
def validate(self) -> bool:
"""
Validate configuration.
Returns:
True if configuration is valid
Raises:
ValueError: If configuration is invalid
"""
# Check primary provider has API key
primary_key = self.get_provider_api_key(self.provider)
if not primary_key:
raise ValueError(f"API key not configured for primary provider: {self.provider}")
# Validate temperature range
if not 0.0 <= self.temperature <= 1.0:
raise ValueError(f"Temperature must be between 0.0 and 1.0, got: {self.temperature}")
# Validate max_tokens
if self.max_tokens <= 0:
raise ValueError(f"max_tokens must be positive, got: {self.max_tokens}")
# Validate conversation limits
if self.max_messages <= 0:
raise ValueError(f"max_messages must be positive, got: {self.max_messages}")
if self.max_conversation_tokens <= 0:
raise ValueError(f"max_conversation_tokens must be positive, got: {self.max_conversation_tokens}")
return True
|