- Upgrade to codestral-latest

Author: Jet Xu

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.REPL.enableREPLSmartSend": false
+}

CHANGELOG.md

@@ -7,6 +7,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [0.2.3] - 2024-11-24
 
+### Optimized
+- Upgrade to codestral-latest
+
+## [0.2.3] - 2024-11-24
+
 ### Optimized
 - Upgrade to mistral-large-2411
 

llama_github/llm_integration/initial_load.py

@@ -1,8 +1,6 @@
 # initial_load.py
 from typing import Optional, Any
 from threading import Lock
-from langchain_openai import ChatOpenAI
-from langchain_mistralai.chat_models import ChatMistralAI
 
 from llama_github.config.config import config
 from llama_github.logger import logger
@@ -67,15 +65,17 @@ def __init__(self,
             self.llm = llm
             self.model_type = "Custom_langchain_llm"
         elif mistral_api_key is not None and mistral_api_key != "" and self.llm is None:
-            logger.info("Initializing Mistral API...")
-            self.llm = ChatMistralAI(mistral_api_key=mistral_api_key, model="mistral-large-2411")
+            logger.info("Initializing Codestral API...")
+            from langchain_mistralai.chat_models import ChatMistralAI
+            self.llm = ChatMistralAI(mistral_api_key=mistral_api_key, model="codestral-latest")
             self.llm_simple = ChatMistralAI(
                 mistral_api_key=mistral_api_key, 
-                model="open-mistral-nemo", 
+                model="open-mistral-nemo",
                 temperature=0.2
             )
             self.model_type = "OpenAI"
         elif openai_api_key is not None and openai_api_key != "" and self.llm is None:
+            from langchain_openai import ChatOpenAI
             logger.info("Initializing OpenAI API...")
             self.llm = ChatOpenAI(api_key=openai_api_key, model="gpt-4-turbo")
             self.llm_simple = ChatOpenAI(

llama_github/version.py

@@ -1 +1 @@
-__version__ = '0.2.6'
+__version__ = '0.2.7'

requirements.txt

@@ -8,7 +8,6 @@ pydantic
 requests
 transformers
 urllib3
-sys
 platform
 subprocess
 pydantic_core

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = llama-github
-version = 0.2.6
+version = 0.2.7
 author = Jet Xu
 author_email = [email protected]
 description = Llama-github is an open-source Python library that empowers LLM Chatbots, AI Agents, and Auto-dev Agents to conduct Retrieval from actively selected GitHub public projects. It Augments through LLMs and Generates context for any coding question, in order to streamline the development of sophisticated AI-driven applications.