Refactor prompt to improve multilingual reponses

sderev · Jun 18, 2023 · 0086e8f · 0086e8f
1 parent 4f48fe3
commit 0086e8f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,5 +1,7 @@
 from setuptools import setup, find_packages
 
+VERSION = "0.1.10"
+
 
 with open("README.md", encoding="UTF-8") as file:
     readme = file.read()
@@ -10,7 +12,7 @@
 setup(
     name="ShellGenius",
     description="ShellGenius is a tool to generate shell commands from description in natural language.",
-    version="0.1.9",
+    version=VERSION,
     packages=find_packages(),
     install_requires=requirements,
     entry_points={

diff --git a/shellgenius/gpt_integration.py b/shellgenius/gpt_integration.py
@@ -9,7 +9,7 @@ def format_prompt(command_description, os_name):
     prompt = [
         {
             "role": "system",
-            "content": f"You are an expert in using {os_name} and the shell terminal.",
+            "content": f"You are an expert AI in using {os_name} and the shell terminal. As an helpful AI, you recognize the language of the user and respond in the same language.",
         },
         {
             "role": "user",