Unified Interface

from src.core.agent_scheduler import RepoMasterAgent
from configs.oai_config import get_llm_config

# Initialize
llm_config = get_llm_config()
code_execution_config = {
    "work_dir": "coding/workspace",
    "use_docker": False
}

agent = RepoMasterAgent(
    llm_config=llm_config,
    code_execution_config=code_execution_config
)

# Use unified interface - automatic agent selection
result = agent.solve_task_with_repo(
    "What is the current stock price of Apple?"
)
print(result)

# Launch unified mode
python launcher.py --mode unified

# Or simply (unified is default)
python launcher.py

def solve_task_with_repo(self, task: str) -> str:
    """
    Enhanced RepoMaster that can work with GitHub repositories, 
    local repositories, or provide general programming assistance.
    
    Three Working Modes:
    1. Web Search Mode - Real-time information retrieval
    2. Repository Mode - Repository analysis and execution  
    3. General Code Assistant Mode - Programming assistance
    """

# Example tasks that trigger Web Search Mode
"What is the current price of Bitcoin?"
"Latest developments in quantum computing"
"Top AI research papers published this week"

# Example tasks that trigger Repository Mode
"Use pdfplumber library to extract PDF text"
"Analyze my local repo at /home/user/project"
"Find and use a GitHub library for image processing"

# Example tasks that trigger General Code Assistant Mode
"Write a function to calculate Fibonacci numbers"
"How do I implement binary search in Python?"
"Debug this code: [code snippet]"

max_turns=12  # Maximum conversation turns
summary_method="reflection_with_llm"  # LLM-based summarization
summary_prompt="Summarize takeaway from the conversation and 
                generate a complete and detailed report at last."

from src.core.conversation_manager import ConversationManager

conversation = ConversationManager(user_id, "unified")
optimized_task = conversation.get_optimized_prompt(task)

llm_config = {
    "cache_seed": 42,  # Enable caching
    # ... other config
}