Nihal2000 commited on
Commit
66cbb2c
·
verified ·
1 Parent(s): 2589ed0

Update services/llm_service.py

Browse files
Files changed (1) hide show
  1. 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 MistralClient
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 = MistralClient(
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")