@@ -79,34 +79,34 @@ def executeSelectAI(cummulativeResult: str = None):
79
79
{"containsWord" : "satellites" , "latestQuestion" : "satellites" , "latestAnswer" : "satellites" },
80
80
{"containsWord" : "spatial" , "latestQuestion" : "spatial" , "latestAnswer" : "spatial" }
81
81
]
82
-
82
+ # other profiles includes AIHOLO , VIDEOGAMES_PROFILE, MOVIESTREAM
83
83
chatquery = """SELECT DBMS_CLOUD_AI.GENERATE(
84
84
prompt => :prompt,
85
- profile_name => 'VIDEOGAMES_PROFILE ',
85
+ profile_name => 'AIHOLO ',
86
86
action => 'chat')
87
87
FROM dual"""
88
88
89
89
narratequery = """SELECT DBMS_CLOUD_AI.GENERATE(
90
90
prompt => :prompt,
91
- profile_name => 'VIDEOGAMES_PROFILE ',
91
+ profile_name => 'AIHOLO ',
92
92
action => 'narrate')
93
93
FROM dual"""
94
94
95
95
runsqlquery = """SELECT DBMS_CLOUD_AI.GENERATE(
96
96
prompt => :prompt,
97
- profile_name => 'VIDEOGAMES_PROFILE ',
97
+ profile_name => 'AIHOLO ',
98
98
action => 'runsql')
99
99
FROM dual"""
100
100
101
101
showssqlquery = """SELECT DBMS_CLOUD_AI.GENERATE(
102
102
prompt => :prompt,
103
- profile_name => 'VIDEOGAMES_PROFILE ',
103
+ profile_name => 'AIHOLO ',
104
104
action => 'showsql')
105
105
FROM dual"""
106
106
107
107
explainsqlquery = """SELECT DBMS_CLOUD_AI.GENERATE(
108
108
prompt => :prompt,
109
- profile_name => 'VIDEOGAMES_PROFILE ',
109
+ profile_name => 'AIHOLO ',
110
110
action => 'explainsql')
111
111
FROM dual"""
112
112
@@ -122,6 +122,9 @@ def executeSelectAI(cummulativeResult: str = None):
122
122
elif isChat :
123
123
query = chatquery
124
124
print ("chat true" )
125
+ elif isRag :
126
+ query = narratequery
127
+ print ("chat true" )
125
128
else :
126
129
query = narratequery
127
130
@@ -283,12 +286,22 @@ async def handle_request(request):
283
286
print (f"cummulativeResult: { cummulativeResult } " )
284
287
285
288
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 } " )
286
297
287
298
if "use rag" in lowered_cumulative_result :
288
299
cummulativeResult = cummulativeResult .replace ("use rag" , "" )
300
+ print (f"use rag issued: { cummulativeResult } " )
289
301
isRag = True
290
- if "use database" in lowered_cumulative_result :
302
+ elif "use database" in lowered_cumulative_result :
291
303
cummulativeResult = cummulativeResult .replace ("use database" , "" )
304
+ print (f"use database issued: { cummulativeResult } " )
292
305
isRag = True
293
306
elif "use chat" in lowered_cumulative_result :
294
307
cummulativeResult = cummulativeResult .replace ("use chat" , "" )
@@ -316,6 +329,7 @@ async def handle_request(request):
316
329
print (f"isRunSQL: { isRunSQL } " )
317
330
print (f"isExplainSQL: { isExplainSQL } " )
318
331
print (f"isRag: { isRag } " )
332
+ print (f"isChat: { isChat } " )
319
333
320
334
if isSelect :
321
335
executeSelectAI (cummulativeResult )
0 commit comments