Sujithanumala commited on
Commit
25cb7e3
·
verified ·
1 Parent(s): d95b2ea

Update Classes/Owiki_Class.py

Browse files
Files changed (1) hide show
  1. Classes/Owiki_Class.py +5 -1
Classes/Owiki_Class.py CHANGED
@@ -32,9 +32,13 @@ class OWiki:
32
  def create_sql_prompt_template(self,schemas):
33
  prompt = """Write an SQL query for the following questions whose schemas are as follows.\nSQL Schema:"""
34
  for table_name,table_schema in schemas.items():
35
- prompt+= f"Table Name: {table_name}, Schema : {table_schema}\n\n"
 
 
36
  prompt+= """\n\nQuestion:{question}\n\nAnswer:"""
 
37
  prompt_template = PromptTemplate(template = prompt,input_variables=['question'])
 
38
  return prompt_template
39
 
40
  def create_prompt_for_OIC_bot(self):
 
32
  def create_sql_prompt_template(self,schemas):
33
  prompt = """Write an SQL query for the following questions whose schemas are as follows.\nSQL Schema:"""
34
  for table_name,table_schema in schemas.items():
35
+ prompt+= f"Table Name: {table_name} Schema:"
36
+ for key,value in table_schema.items():
37
+ prompt+= f"{key} {value}"
38
  prompt+= """\n\nQuestion:{question}\n\nAnswer:"""
39
+ # print("Prompt",prompt,'\n\n')
40
  prompt_template = PromptTemplate(template = prompt,input_variables=['question'])
41
+ # print("Prompt template",prompt_template)
42
  return prompt_template
43
 
44
  def create_prompt_for_OIC_bot(self):