@@ -79,34 +79,34 @@ def executeSelectAI(cummulativeResult: str = None):
7979 {"containsWord" : "satellites" , "latestQuestion" : "satellites" , "latestAnswer" : "satellites" },
8080 {"containsWord" : "spatial" , "latestQuestion" : "spatial" , "latestAnswer" : "spatial" }
8181 ]
82-
82+ # other profiles includes AIHOLO , VIDEOGAMES_PROFILE, MOVIESTREAM
8383 chatquery = """SELECT DBMS_CLOUD_AI.GENERATE(
8484 prompt => :prompt,
85- profile_name => 'VIDEOGAMES_PROFILE ',
85+ profile_name => 'AIHOLO ',
8686 action => 'chat')
8787 FROM dual"""
8888
8989 narratequery = """SELECT DBMS_CLOUD_AI.GENERATE(
9090 prompt => :prompt,
91- profile_name => 'VIDEOGAMES_PROFILE ',
91+ profile_name => 'AIHOLO ',
9292 action => 'narrate')
9393 FROM dual"""
9494
9595 runsqlquery = """SELECT DBMS_CLOUD_AI.GENERATE(
9696 prompt => :prompt,
97- profile_name => 'VIDEOGAMES_PROFILE ',
97+ profile_name => 'AIHOLO ',
9898 action => 'runsql')
9999 FROM dual"""
100100
101101 showssqlquery = """SELECT DBMS_CLOUD_AI.GENERATE(
102102 prompt => :prompt,
103- profile_name => 'VIDEOGAMES_PROFILE ',
103+ profile_name => 'AIHOLO ',
104104 action => 'showsql')
105105 FROM dual"""
106106
107107 explainsqlquery = """SELECT DBMS_CLOUD_AI.GENERATE(
108108 prompt => :prompt,
109- profile_name => 'VIDEOGAMES_PROFILE ',
109+ profile_name => 'AIHOLO ',
110110 action => 'explainsql')
111111 FROM dual"""
112112
@@ -122,6 +122,9 @@ def executeSelectAI(cummulativeResult: str = None):
122122 elif isChat :
123123 query = chatquery
124124 print ("chat true" )
125+ elif isRag :
126+ query = narratequery
127+ print ("chat true" )
125128 else :
126129 query = narratequery
127130
@@ -283,12 +286,22 @@ async def handle_request(request):
283286 print (f"cummulativeResult: { cummulativeResult } " )
284287
285288 lowered_cumulative_result = cummulativeResult .lower ()
289+
290+ print (f"Current cummulative result before trimming `use xyz`: { cummulativeResult } " )
291+ print (f"isSelect: { isSelect } " )
292+ print (f"isShowSQL: { isShowSQL } " )
293+ print (f"isRunSQL: { isRunSQL } " )
294+ print (f"isExplainSQL: { isExplainSQL } " )
295+ print (f"isRag: { isRag } " )
296+ print (f"isChat: { isChat } " )
286297
287298 if "use rag" in lowered_cumulative_result :
288299 cummulativeResult = cummulativeResult .replace ("use rag" , "" )
300+ print (f"use rag issued: { cummulativeResult } " )
289301 isRag = True
290- if "use database" in lowered_cumulative_result :
302+ elif "use database" in lowered_cumulative_result :
291303 cummulativeResult = cummulativeResult .replace ("use database" , "" )
304+ print (f"use database issued: { cummulativeResult } " )
292305 isRag = True
293306 elif "use chat" in lowered_cumulative_result :
294307 cummulativeResult = cummulativeResult .replace ("use chat" , "" )
@@ -316,6 +329,7 @@ async def handle_request(request):
316329 print (f"isRunSQL: { isRunSQL } " )
317330 print (f"isExplainSQL: { isExplainSQL } " )
318331 print (f"isRag: { isRag } " )
332+ print (f"isChat: { isChat } " )
319333
320334 if isSelect :
321335 executeSelectAI (cummulativeResult )
0 commit comments