Update services/llm_service.py
Browse files- services/llm_service.py +2 -2
 
    	
        services/llm_service.py
    CHANGED
    
    | 
         @@ -2,7 +2,7 @@ import logging 
     | 
|
| 2 | 
         
             
            import asyncio
         
     | 
| 3 | 
         
             
            from typing import List, Dict, Any, Optional
         
     | 
| 4 | 
         
             
            import anthropic
         
     | 
| 5 | 
         
            -
            from mistralai.client import  
     | 
| 6 | 
         
             
            import config
         
     | 
| 7 | 
         | 
| 8 | 
         
             
            logger = logging.getLogger(__name__)
         
     | 
| 
         @@ -27,7 +27,7 @@ class LLMService: 
     | 
|
| 27 | 
         
             
                            logger.info("Anthropic client initialized")
         
     | 
| 28 | 
         | 
| 29 | 
         
             
                        if self.config.MISTRAL_API_KEY:
         
     | 
| 30 | 
         
            -
                            self.mistral_client =  
     | 
| 31 | 
         
             
                                api_key=self.config.MISTRAL_API_KEY
         
     | 
| 32 | 
         
             
                            )
         
     | 
| 33 | 
         
             
                            logger.info("Mistral client initialized")
         
     | 
| 
         | 
|
| 2 | 
         
             
            import asyncio
         
     | 
| 3 | 
         
             
            from typing import List, Dict, Any, Optional
         
     | 
| 4 | 
         
             
            import anthropic
         
     | 
| 5 | 
         
            +
            from mistralai.client import Mistral
         
     | 
| 6 | 
         
             
            import config
         
     | 
| 7 | 
         | 
| 8 | 
         
             
            logger = logging.getLogger(__name__)
         
     | 
| 
         | 
|
| 27 | 
         
             
                            logger.info("Anthropic client initialized")
         
     | 
| 28 | 
         | 
| 29 | 
         
             
                        if self.config.MISTRAL_API_KEY:
         
     | 
| 30 | 
         
            +
                            self.mistral_client = Mistral(
         
     | 
| 31 | 
         
             
                                api_key=self.config.MISTRAL_API_KEY
         
     | 
| 32 | 
         
             
                            )
         
     | 
| 33 | 
         
             
                            logger.info("Mistral client initialized")
         
     |