version info reading method updated, test updated

Author: gitdevjin

README.md

@@ -5,7 +5,7 @@
 **CodeMage** is a tool that translates a source file written in one programming language into another language.
 The translation will be done by Large Language Model AI(such as ChatGPT)
 
-## Release 0.1
+## Release 1.0.0
 
 ### Features
 
@@ -14,7 +14,54 @@ The translation will be done by Large Language Model AI(such as ChatGPT)
     3. Supported LLM model: openrouter, groq
     4. Default LLM model is openrouter(sao10k/l3-euryale-70b)
 
-# Getting Started
+# Getting Started with `PyPI`
+
+### 1. Download the package using PyPI
+```bash
+pip install code-mage
+```
+
+### 2. Create your API_KEY at [openrouter](https://openrouter.ai/docs/api-keys) Or [Groq](https://console.groq.com/keys)
+
+It's free with sign-up. You can easily sign-up with your google account
+
+### 3.a Set-up the API_KEY using `Variable` (option 1)
+
+```bash
+export API_KEY=YOUR-API-KEY
+```
+
+### 3.b Set-up with `.toml` file (option 2)
+
+With this option, you can also set other tool options.
+
+- Start by creating a `.codemage-config.toml` in your home directory.
+
+```bash
+mkdir ~/.codemage-config.toml
+```
+
+- Add the following information to the file:
+
+```toml
+model="groq" # if you wish to use OPEN ROUTER you can just delete this line
+GROQ_API_KEY="<YOUR-GROQ-API-KEY>"
+OPENROUTER_API_KEY="<YOUR-OPEN-ROUTER-API-KEY>"
+language="java" # you can use any of the supported languages
+stream=false # if you wish to get the output streamed, set it `true`
+token_usage=false # if you wish to get token_usage info, set it `true`
+output="result" # type any name for the output file without the extension
+```
+
+### 4. Run the tool
+
+```bash
+codemage <file-you-want-to-convert>
+# For example,
+codemage test.js
+```
+
+# Getting Started by cloning the github repo
 
 ### 1. Install Python : https://www.python.org/downloads/
 

code_mage/argsParser.py

@@ -2,7 +2,7 @@
 
 
 def arg_parser(config):
-	VERSION = "release 0.1"
+	VERSION = config.get("version")
 
 	parser = argparse.ArgumentParser(
 		description="This Tool translates a source file into another programming language."

code_mage/loadConfig.py

@@ -10,4 +10,12 @@ def load_config():
 	else:
 		config = {}
 
+	config["version"] = __get_version_from_pyproject()
+
 	return config
+
+
+def __get_version_from_pyproject():
+	with open("pyproject.toml", "r") as f:
+		pyproject_data = toml.load(f)
+	return pyproject_data["tool"]["poetry"]["version"]

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-mage"
-version = "0.9.0"
+version = "1.0.0"
 description = ""
 authors = ["gitdevjin <[email protected]>"]
 readme = "README.md"

tests/test_loadConfig.py

@@ -1,37 +1,74 @@
 import os
-from unittest.mock import patch
+from unittest.mock import patch, mock_open
 from code_mage.loadConfig import load_config  # Adjust the import path accordingly
 
 
 class TestLoadConfig:
 	@patch("os.path.exists")
 	@patch("toml.load")
-	def test_load_config_file_exists(self, mock_toml_load, mock_os_exists):
-		# Simulate that the file exists
+	@patch("builtins.open", new_callable=mock_open)
+	def test_load_config_file_exists(self, mock_file, mock_toml_load, mock_os_exists):
+		# Simulate that the config file exists
 		mock_os_exists.return_value = True
 
-		# Simulate the contents of the file
-		mock_toml_load.return_value = {
-			"language": "Python",
-			"OPENROUTER_API_KEY": "mock_openrouter_api_key",
-		}
+		# Simulate the contents of the config file
+		mock_toml_load.side_effect = [
+			{
+				"language": "Python",
+				"OPENROUTER_API_KEY": "mock_openrouter_api_key",
+			},  # First call (config file)
+			{"tool": {"poetry": {"version": "0.9.0"}}},  # Second call (pyproject.toml)
+		]
 
 		# Call load_config function
 		config = load_config()
 
 		# Assertions
 		mock_os_exists.assert_called_once_with(os.path.expanduser("~/.codemage-config.toml"))
-		mock_toml_load.assert_called_once_with(os.path.expanduser("~/.codemage-config.toml"))
-		assert config == {"language": "Python", "OPENROUTER_API_KEY": "mock_openrouter_api_key"}
+		mock_file.assert_called_once_with("pyproject.toml", "r")
+		assert config == {
+			"language": "Python",
+			"OPENROUTER_API_KEY": "mock_openrouter_api_key",
+			"version": "0.9.0",
+		}
 
 	@patch("os.path.exists")
-	def test_load_config_file_not_exists(self, mock_os_exists):
+	@patch("toml.load")
+	@patch("builtins.open", new_callable=mock_open)
+	def test_load_config_file_with_no_content(self, mock_file, mock_toml_load, mock_os_exists):
 		# Simulate that the file does not exist
+		mock_os_exists.return_value = True
+
+		# Call the function under test
+
+		mock_toml_load.side_effect = [
+			{},
+			{"tool": {"poetry": {"version": "0.9.0"}}},  # Second call (pyproject.toml)
+		]
+
+		config = load_config()
+
+		# Assertions
+		mock_os_exists.assert_called_once_with(os.path.expanduser("~/.codemage-config.toml"))
+		mock_file.assert_called_once_with("pyproject.toml", "r")
+
+		# Assert the default config with version added
+		assert config == {"version": "0.9.0"}
+
+	@patch("os.path.exists")  # Third argument
+	@patch("builtins.open")  # Second argument
+	@patch("toml.load")  # First argument
+	def test_load_config_file_not_exists(self, mock_toml_load, mock_file, mock_os_exists):
+		# Simulate that the config file does not exist
 		mock_os_exists.return_value = False
 
+		# Simulate the content of the pyproject.toml file
+		mock_toml_load.return_value = {"tool": {"poetry": {"version": "0.9.0"}}}
+
 		# Call the function under test
 		config = load_config()
 
 		# Assertions
 		mock_os_exists.assert_called_once_with(os.path.expanduser("~/.codemage-config.toml"))
-		assert config == {}
+		mock_file.assert_called_once_with("pyproject.toml", "r")
+		assert config == {"version": "0.9.0"}