crennis/openai-sql
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed GPT

Browse Source
This commit is contained in:
crennis 2023-05-04 10:53:21 +02:00
parent 13e027afdf
commit 2a2cb0d507
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nogui/modules/aisql.py
View File

@ -8,7 +8,9 @@ class AI:
self.convertlog = [] self.convertlog = []
def humantosql(self, text: str, dbtype: str, tableschema: list) -> str: def humantosql(self, text: str, dbtype: str, tableschema: list) -> str:
self.convertlog = [{"role": "system", "content": f"You convert Human Language to SQL. Only answer as an SQL Statement as the output of you will be the direct input into the database. Always edit the old Statement and do not create a completly new one. You are using this Database: {dbtype} and for better context here are the tables and columns: {tableschema}"}] if not self.convertlog:
self.convertlog = [{"role": "system", "content": f"You convert Human Language to SQL. Only answer as an Valid SQL Statement. Always modify your previous answer and do not create something new. You are using this Database: {dbtype}. For better context here are the tables and columns: {tableschema}"}]
prompt = {"role": "user", "content": text} prompt = {"role": "user", "content": text}
self.convertlog.append(prompt) self.convertlog.append(prompt)
response = openai.ChatCompletion.create( response = openai.ChatCompletion.create(
@ -17,6 +19,9 @@ class AI:
) )
response = response['choices'][0]['message']['content'] response = response['choices'][0]['message']['content']
self.convertlog.append({"role": "system", "content": response}) self.convertlog.append({"role": "system", "content": response})
for i in self.convertlog:
print(i)
return response return response
def decide(self, sql: str) -> str: def decide(self, sql: str) -> str: