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

Fixed a SQL-Validity Check

Browse Source
This commit is contained in:
crennis 2023-05-04 11:39:01 +02:00
parent d72a0863a3
commit 9d76648a7d
1 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
nogui/modules/aisql.py
View File

@ -6,7 +6,7 @@ class AI:
self.api_key = api_key self.api_key = api_key
openai.api_key = self.api_key openai.api_key = self.api_key
self.convertlog = [] self.convertlog = []
self.response = None
def humantosql(self, text: str, dbtype: str, tableschema: list) -> str: def humantosql(self, text: str, dbtype: str, tableschema: list) -> str:
if not self.convertlog: if not self.convertlog:
@ -21,18 +21,11 @@ class AI:
self.response = self.response['choices'][0]['message']['content'] self.response = self.response['choices'][0]['message']['content']
self.convertlog.append({"role": "assistant", "content": self.response}) self.convertlog.append({"role": "assistant", "content": self.response})
for i in self.convertlog:
print(i)
# FIXME: After the first validation failed it will always return None
print(self.response)
if self.validate(self.response) is True: if self.validate(self.response) is True:
return self.response return self.response
else: else:
self.humantosql("Only answer as a Valid SQL-Statement don't add any additional text", dbtype, tableschema) valid = self.humantosql("Only answer as a Valid SQL-Statement don't add any additional text", dbtype, tableschema)
return valid
def decide(self, sql: str) -> str: def decide(self, sql: str) -> str:
prompt = [{"role": "system", "content": "You have to decide which function it should use. Answer with [FETCHALL] to fetch all, [FETCHONE] to fetch only one, [FETCHMANY=N] to fetchmany with N being the range, [EXECUTE] to just execute, [EXECUTEMANY=N] to execute many with N being the range"}, prompt = [{"role": "system", "content": "You have to decide which function it should use. Answer with [FETCHALL] to fetch all, [FETCHONE] to fetch only one, [FETCHMANY=N] to fetchmany with N being the range, [EXECUTE] to just execute, [EXECUTEMANY=N] to execute many with N being the range"},