Version 2.3.0 Added Deepseek V3/R1 and OpenAI O1 model support

haseeb-heaven · haseeb-heaven · commit 50acc69bb432 · 2025-02-02T02:35:13.000+05:30
diff --git a/README.md b/README.md
@@ -99,8 +99,9 @@ export HUGGINGFACE_API_KEY="Your HuggingFace API Key"
 export PALM_API_KEY="Your Google Palm API Key"
 export GEMINI_API_KEY="Your Google Gemini API Key"
 export OPENAI_API_KEY="Your OpenAI API Key"
-export GROQ_API_KEY="Your Groq AI API Key"
-export ANTHROPIC_API_KEY="Your Anthropic AI API Key"
+export GROQ_API_KEY="Your Groq API Key"
+export ANTHROPIC_API_KEY="Your Anthropic API Key"
+export DEEPSEEK_API_KEY="Your Deepseek API Key"
 ```
 
 # Offline models setup.</br>
@@ -169,6 +170,11 @@ To use Code-Interpreter, use the following command options:
 - List of all **models** are (**Contribute - MORE**): </br>
     - `gpt-3.5-turbo` - Generates code using the GPT 3.5 Turbo model.
     - `gpt-4` - Generates code using the GPT 4 model.
+	- `o1-mini` - Generates code using the OpenAI o1-mini model.
+	- `o1-preview` - Generates code using the OpenAI o1-preview model.
+	- `deepseek-chat` - Generates response using the Deepseek chat model.
+	- `deepseek-coder` - Generates code using the Deepseek coder model.
+	- `deepseek-reasoner` - Generates code using the Deepseek reasoner model.
     - `gemini-pro` - Generates code using the Gemini Pro model.
     - `palm-2` - Generates code using the PALM 2 model.
     - `claude-2` - Generates code using the AnthropicAI Claude-2 model.
@@ -337,6 +343,8 @@ If you're interested in contributing to **Code-Interpreter**, we'd love to have
   - Resolved pip package installation issues for smoother and more reliable setup.
 - **v2.2.1** - Fixed **No Content/Response from LLM** Bug, Fixed _Debug Mode_ with **Logs**.
 
+- **v2.3.0** - Added Deepseek V3 and R1 models support now. Added OpenAI o1 Models support.
+
 ---
 
 ## 📜 **License**
diff --git a/configs/deepseek-chat.config b/configs/deepseek-chat.config
@@ -0,0 +1,6 @@
+temperature=0.1
+max_tokens=1024
+start_sep=```
+end_sep=```
+skip_first_line=False
+HF_MODEL = deepseek-chat
diff --git a/configs/deepseek-coder.config b/configs/deepseek-coder.config
@@ -0,0 +1,6 @@
+temperature=0.1
+max_tokens=1024
+start_sep=```
+end_sep=```
+skip_first_line=False
+HF_MODEL = deepseek-coder
diff --git a/configs/deepseek-reasoner.config b/configs/deepseek-reasoner.config
@@ -0,0 +1,6 @@
+temperature=0.1
+max_tokens=1024
+start_sep=```
+end_sep=```
+skip_first_line=False
+HF_MODEL = deepseek-reasoner
diff --git a/configs/gpt-o1-mini.config b/configs/gpt-o1-mini.config
@@ -0,0 +1,18 @@
+
+# The temperature parameter controls the randomness of the model's output. Lower values make the output more deterministic.
+temperature = 0.1
+
+# The maximum number of new tokens that the model can generate.
+max_tokens = 1024
+
+# The start separator for the generated code.
+start_sep = ```
+
+# The end separator for the generated code.
+end_sep = ```
+
+# If True, the first line of the generated text will be skipped.
+skip_first_line = True
+
+# The model used for generating the code.
+HF_MODEL = o1-mini
diff --git a/configs/gpt-o1-preview.config b/configs/gpt-o1-preview.config
@@ -0,0 +1,18 @@
+
+# The temperature parameter controls the randomness of the model's output. Lower values make the output more deterministic.
+temperature = 0.1
+
+# The maximum number of new tokens that the model can generate.
+max_tokens = 1024
+
+# The start separator for the generated code.
+start_sep = ```
+
+# The end separator for the generated code.
+end_sep = ```
+
+# If True, the first line of the generated text will be skipped.
+skip_first_line = True
+
+# The model used for generating the code.
+HF_MODEL = o1-preview
diff --git a/interpreter b/interpreter
@@ -13,7 +13,7 @@ Command line arguments:
 --display_code, -dc: Displays the generated code in the output.
 
 Author: HeavenHM
-Date: 2023/12/01
+Date: 2025/01/01
 """
 
 from libs.interpreter_lib import Interpreter
@@ -25,57 +25,58 @@ from libs.markdown_code import display_markdown_message
 from libs.utility_manager import UtilityManager
 
 # The main version of the interpreter.
-INTERPRETER_VERSION = "2.1.3"
+INTERPRETER_VERSION = "2.3.0"
+
 
 def main():
-        parser = argparse.ArgumentParser(description='Code - Interpreter')
-        parser.add_argument('--exec', '-e', action='store_true', default=False, help='Execute the code')
-        parser.add_argument('--save_code', '-s', action='store_true', default=False, help='Save the generated code')
-        parser.add_argument('--mode', '-md', choices=['code', 'script', 'command','vision','chat'], help='Select the mode (`code` for generating code, `script` for generating shell scripts, `command` for generating single line commands) `vision` for generating text from images')
-        parser.add_argument('--model', '-m', type=str, default='code-llama', help='Set the model for code generation. (Defaults to gpt-3.5-turbo)')
-        parser.add_argument('--version', '-v', action='version', version='%(prog)s '+ INTERPRETER_VERSION)
-        parser.add_argument('--lang', '-l', type=str, default='python', help='Set the interpreter language. (Defaults to Python)')
-        parser.add_argument('--display_code', '-dc', action='store_true', default=False, help='Display the code in output')
-        parser.add_argument('--history', '-hi', action='store_true', default=False, help='Use history as memory')
-        parser.add_argument('--upgrade', '-up', action='store_true', default=False, help='Upgrade the interpreter')
-        parser.add_argument('--file', '-f', type=str, nargs='?', const='prompt.txt', default=None, help='Sets the file to read the input prompt from')
-        args = parser.parse_args()
+	parser = argparse.ArgumentParser(description='Code - Interpreter')
+	parser.add_argument('--exec', '-e', action='store_true', default=False, help='Execute the code')
+	parser.add_argument('--save_code', '-s', action='store_true', default=False, help='Save the generated code')
+	parser.add_argument('--mode', '-md', choices=['code', 'script', 'command', 'vision', 'chat'], help='Select the mode (`code` for generating code, `script` for generating shell scripts, `command` for generating single line commands) `vision` for generating text from images')
+	parser.add_argument('--model', '-m', type=str, default='code-llama', help='Set the model for code generation. (Defaults to gpt-3.5-turbo)')
+	parser.add_argument('--version', '-v', action='version', version='%(prog)s ' + INTERPRETER_VERSION)
+	parser.add_argument('--lang', '-l', type=str, default='python', help='Set the interpreter language. (Defaults to Python)')
+	parser.add_argument('--display_code', '-dc', action='store_true', default=False, help='Display the code in output')
+	parser.add_argument('--history', '-hi', action='store_true', default=False, help='Use history as memory')
+	parser.add_argument('--upgrade', '-up', action='store_true', default=False, help='Upgrade the interpreter')
+	parser.add_argument('--file', '-f', type=str, nargs='?', const='prompt.txt', default=None, help='Sets the file to read the input prompt from')
+	args = parser.parse_args()
 
-        # Check if only the application name is passed
-        if len(sys.argv) <= 1:
-            parser.print_help()
-            return
-        
-        warnings.filterwarnings("ignore")  # To ignore all warnings
+	# Check if only the application name is passed
+	if len(sys.argv) <= 1:
+		parser.print_help()
+		return
+	
+	warnings.filterwarnings("ignore")  # To ignore all warnings
 
-        # Upgrade the interpreter if the --upgrade flag is passed.
-        if args.upgrade:
-            UtilityManager.upgrade_interpreter()
-            return
-        
-        # Create an instance of the Interpreter class and call the main method.
-        interpreter = Interpreter(args)
-        interpreter.interpreter_main(INTERPRETER_VERSION)
-        
-if __name__ == "__main__":
-    try:
-        main()
-    except SystemExit:
-        pass  # Ignore the SystemExit exception caused by --version argument
-    except Exception as exception:
+	# Upgrade the interpreter if the --upgrade flag is passed.
+	if args.upgrade:
+		UtilityManager.upgrade_interpreter()
+		return
+	
+	# Create an instance of the Interpreter class and call the main method.
+	interpreter = Interpreter(args)
+	interpreter.interpreter_main(INTERPRETER_VERSION)
 
-        # Print a meaningful error message if the interpreter is not setup properly.
-        if ".env file" in str(exception):
-            display_markdown_message("Interpreter is not setup properly. Please follow these steps \
-to setup the interpreter:\n\
-1. Create a .env file in the root directory of the project.\n\
-2. Add the following line to the .env file:\n\
-GEMINI_API_KEY=<your api key>\n\
-OPENAI_API_KEY=<your api key>\n\
-ANTHROPIC_API_KEY=<your api key>\n\
-3. Replace <your api key> with your OpenAI/Gemini API key.\n\
-4. Run the interpreter again.")
-            
-        else:
-            display_markdown_message(f"An error occurred: {exception}")
-            traceback.print_exc()
+
+if __name__ == "__main__":
+	try:
+		main()
+	except SystemExit:
+		pass  # Ignore the SystemExit exception caused by --version argument
+	except Exception as exception:
+		# Print a meaningful error message if the interpreter is not setup properly.
+		if ".env file" in str(exception):
+			display_markdown_message("Interpreter is not setup properly. Please follow these steps \
+			to setup the interpreter:\n\
+			1. Create a .env file in the root directory of the project.\n\
+			2. Add the following line to the .env file:\n\
+			GEMINI_API_KEY=<your api key>\n\
+			OPENAI_API_KEY=<your api key>\n\
+			ANTHROPIC_API_KEY=<your api key>\n\
+			3. Replace <your api key> with your OpenAI/Gemini API key.\n\
+			4. Run the interpreter again.")
+			
+		else:
+			display_markdown_message(f"An error occurred interpreter main: {exception}")
+			traceback.print_exc()
diff --git a/interpreter.py b/interpreter.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-*
 """
 This is the main file for the Code-Interpreter.
 It handles command line arguments and initializes the Interpreter.
@@ -24,7 +25,7 @@
 from libs.utility_manager import UtilityManager
 
 # The main version of the interpreter.
-INTERPRETER_VERSION = "2.2.1"
+INTERPRETER_VERSION = "2.3.0"
 
 
 def main():
@@ -64,18 +65,17 @@ def main():
 	except SystemExit:
 		pass  # Ignore the SystemExit exception caused by --version argument
 	except Exception as exception:
-
 		# Print a meaningful error message if the interpreter is not setup properly.
 		if ".env file" in str(exception):
 			display_markdown_message("Interpreter is not setup properly. Please follow these steps \
-to setup the interpreter:\n\
-1. Create a .env file in the root directory of the project.\n\
-2. Add the following line to the .env file:\n\
-GEMINI_API_KEY=<your api key>\n\
-OPENAI_API_KEY=<your api key>\n\
-ANTHROPIC_API_KEY=<your api key>\n\
-3. Replace <your api key> with your OpenAI/Gemini API key.\n\
-4. Run the interpreter again.")
+			to setup the interpreter:\n\
+			1. Create a .env file in the root directory of the project.\n\
+			2. Add the following line to the .env file:\n\
+			GEMINI_API_KEY=<your api key>\n\
+			OPENAI_API_KEY=<your api key>\n\
+			ANTHROPIC_API_KEY=<your api key>\n\
+			3. Replace <your api key> with your OpenAI/Gemini API key.\n\
+			4. Run the interpreter again.")
 			
 		else:
 			display_markdown_message(f"An error occurred interpreter main: {exception}")
diff --git a/libs/code_interpreter.py b/libs/code_interpreter.py
@@ -216,3 +216,4 @@ def execute_command(self, command:str):
 		except Exception as exception:
 			self.logger.error(f"Error in executing command: {str(exception)}")
 			raise exception
+
diff --git a/libs/interpreter_lib.py b/libs/interpreter_lib.py
@@ -25,7 +25,6 @@
 from dotenv import load_dotenv
 import shlex
 import shutil
-import logging
 
 class Interpreter:
 	logger = None
@@ -100,9 +99,10 @@ def initialize(self):
 			self.logger.error("Exception on initializing readline history")
 
 	def initialize_client(self):
-		load_dotenv()
+		load_dotenv(dotenv_path=os.path.join(os.getcwd(), "../.env"), override=True)
 		self.logger.info("Initializing Client")
-		
+		config_file_name: str = ""
+
 		self.logger.info(f"Interpreter model selected is '{self.INTERPRETER_MODEL}'")
 		if self.INTERPRETER_MODEL is None or self.INTERPRETER_MODEL == "":
 			self.logger.info("HF_MODEL is not provided, using default model.")
@@ -126,7 +126,7 @@ def initialize_client(self):
 				self.logger.info("Using local API key from environment variables.")
 				
 			if api_key is None:
-				load_dotenv(dotenv_path=os.path.join(os.getcwd(), ".env"))
+				load_dotenv(dotenv_path=os.path.join(os.getcwd(), "../.env"), override=True)
 				api_key = os.getenv('OPENAI_API_KEY')
 				if api_key is None:
 					self.logger.info("Setting default local API key for local models.")
@@ -141,6 +141,7 @@ def initialize_client(self):
 			"claude": {"key_name": "ANTHROPIC_API_KEY", "prefix": "sk-ant-"},
 			"palm": {"key_name": "PALM_API_KEY", "prefix": None, "length": 15},
 			"gemini": {"key_name": "GEMINI_API_KEY", "prefix": None, "length": 15},
+			"deepseek": {"key_name": "DEEPSEEK_API_KEY", "prefix": None, "length": 10},
 			"default": {"key_name": "HUGGINGFACE_API_KEY", "prefix": "hf_"}
 		}
 
@@ -149,7 +150,7 @@ def initialize_client(self):
 				api_key_name = api_key_info["key_name"]
 				api_key = os.getenv(api_key_name)
 				if api_key is None:
-					load_dotenv(dotenv_path=os.path.join(os.getcwd(), ".env"))
+					load_dotenv(dotenv_path=os.path.join(os.getcwd(), "../.env"), override=True)
 					api_key = os.getenv(api_key_name)
 				if not api_key:
 					raise Exception(f"{api_key_name} not found in .env file.")
@@ -168,8 +169,8 @@ def initialize_mode(self):
 		if not self.SCRIPT_MODE and not self.COMMAND_MODE and not self.VISION_MODE and not self.CHAT_MODE:
 			self.CODE_MODE = True
 	
-	def get_prompt(self, message: str, chat_history: List[dict]) -> str:
-		system_message = None
+	def get_prompt(self, message: str, chat_history: List[dict]) -> List[dict] | str:
+		system_message: str = ""
 		
 		if self.CODE_MODE:
 			system_message = self.system_message
@@ -357,6 +358,14 @@ def generate_content(self, message, chat_history: list[tuple[str, str]], tempera
 				raise Exception("Exception api base not set for custom model")
 			self.logger.info("Response received from completion function.")
 
+		# Check if the model is Deepseek
+		elif 'deepseek' in self.INTERPRETER_MODEL:
+			self.logger.info("Model is Deepseek.")
+			 # Ensure the model string is prefixed with "deepseek/"
+			if not self.INTERPRETER_MODEL.startswith("deepseek/"):
+				self.INTERPRETER_MODEL = "deepseek/" + self.INTERPRETER_MODEL
+			response = litellm.completion(self.INTERPRETER_MODEL, messages=messages, temperature=temperature, max_tokens=max_tokens)
+			self.logger.info("Response received from Deepseek completion.")
 
 		# Check if model are from Hugging Face.
 		else:
@@ -421,7 +430,7 @@ def get_command_prompt(self,  task, os_name):
 			"NOTE: Ensure the command is compatible with the specified OS and version.\n"
 			"Output should only contain the command, with no additional text."
 		)
-		self.logger.info(f"Command Prompt: {prompt}")
+		self.logger.info("Command Prompt: {prompt}")
 		return prompt
 
 	def handle_vision_mode(self,  task):
@@ -764,7 +773,7 @@ def interpreter_main(self,  version):
 						continue
 
 					fix_prompt = f"Fix the errors in {self.INTERPRETER_LANGUAGE} language.\nCode is \n'{code_snippet}'\nAnd Error is \n'{code_error}'\n"
-					f"give me output only in code and no other text or explanation. And comment in code where you fixed the error.\n"
+					"give me output only in code and no other text or explanation. And comment in code where you fixed the error.\n"
 					
 					# Start the LLM Request.
 					self.logger.info(f"Fix Prompt: {fix_prompt}")
diff --git a/libs/logger.py b/libs/logger.py
@@ -1,14 +1,14 @@
 import logging
 from logging.handlers import RotatingFileHandler
-
+from typing import Optional
 
 class Logger:
-	_logger = None
+	_logger: Optional[logging.Logger] = None
 	_file_handler = None
 	_console_handler = None
 
 	@staticmethod
-	def initialize(filename: str):
+	def initialize(filename: str) -> logging.Logger:
 		if Logger._logger is None:
 			Logger._logger = logging.getLogger(filename)
 			Logger._logger.setLevel(logging.DEBUG)
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,6 @@
 # Libraries for code interpretation and execution
-litellm  
+litellm>=1.60.0
+litellm[proxy]
 Pygments 
 pyreadline 
 python-dotenv 
@@ -15,3 +16,6 @@ plotly
 
 # Libraries for standard libries 
 stdlib_list
+
+# Legacy libraries
+legacy-cgi
