llm_interface.py

import os
import anthropic
from typing import Optional, List, Dict, Any
from dotenv import load_dotenv
from config import Config

class LLMInterface:
    """Interface for communicating with Anthropic's Claude API."""
    
    def __init__(self, prompts_dir: str = None):
        # Load environment variables from .env file
        load_dotenv()
        
        # Get API key from environment
        api_key = os.getenv("ANTHROPIC_API_KEY")
        if not api_key:
            raise ValueError("ANTHROPIC_API_KEY environment variable is required")
        self.client = anthropic.Anthropic(api_key=api_key)
        if prompts_dir is None:
            self.prompts_dir = os.path.join(os.path.dirname(__file__), 'prompts')
        else:
            self.prompts_dir = prompts_dir
        self.system_prompt = self._load_prompt("system_constitution")

    def _load_prompt(self, prompt_name: str):
        """Loads a prompt from a .txt file in the prompts directory."""
        prompt_path = os.path.join(self.prompts_dir, f"{prompt_name}.txt")
        try:
            with open(prompt_path, 'r') as f:
                return f.read()
        except FileNotFoundError:
            raise ValueError(f"Prompt file not found at: {prompt_path}")

    def get_response(self, prompt_name: str, images: Optional[List[str]] = None, 
                    placeholders: Optional[Dict[str, Any]] = None) -> Optional[str]:
        """
        Sends a request to the LLM using a prompt from a .txt file.
        
        Args:
            prompt_name: Name of the prompt file (without .txt extension)
            images: List of base64 encoded image strings
            placeholders: Dictionary of placeholder values for prompt formatting
            
        Returns:
            The LLM response text, or None if an error occurred
            
        Raises:
            LLMError: If the LLM request fails
        """
        user_prompt_template = self._load_prompt(prompt_name)

        if placeholders:
            user_prompt = user_prompt_template.format(**placeholders)
        else:
            user_prompt = user_prompt_template

        content = [{"type": "text", "text": user_prompt}]
        if images:
            for img_data in images:
                content.insert(0, {
                    "type": "image",
                    "source": {
                        "type": "base64",
                        "media_type": "image/png",
                        "data": img_data,
                    },
                })
        try:
            message = self.client.messages.create(
                model=Config.CLAUDE_MODEL,
                max_tokens=Config.MAX_TOKENS,
                system=self.system_prompt,
                messages=[{"role": "user", "content": content}],
            )
            return message.content[0].text
        except Exception as e:
            print(f"An error occurred: {e}")
            return None

if __name__ == '__main__':
    print("LLMInterface class updated to use individual .txt files from the 'prompts/' directory.")