Ufoptg commited on
Commit
6cd4ef0
1 Parent(s): 8c7269f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -317,7 +317,7 @@ def sibyl_system(
317
  return {"status": "false", "message": "Not Found User"}
318
 
319
  @app.post("/UFoP/gemini-the-oracle", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
320
- def gemini_oracle(item: GeminiPro):
321
  if item.is_multi_chat:
322
  selected_api_key = ASSISTANT_GOOGLE_API_KEYS or item.gemini_api_key
323
  oracle_base = ORACLE_TOKEN or item.oracle_base
@@ -331,10 +331,10 @@ def gemini_oracle(item: GeminiPro):
331
  )
332
  cybersecuritydb(item.user_id, item.mongo_url)
333
  if item.oracle_base == "Delete":
334
- clearedhistory = geni._clear_oracle_history_in_db()
335
  return SuccessResponse(status="True", randydev={"message": f"Oracle Status: {clearedhistory}"})
336
  else:
337
- answer, oracle_chat = geni._GeminiLatest__get_response_oracle(item.query)
338
  return SuccessResponse(
339
  status="True",
340
  randydev={
 
317
  return {"status": "false", "message": "Not Found User"}
318
 
319
  @app.post("/UFoP/gemini-the-oracle", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
320
+ async def gemini_oracle(item: GeminiPro):
321
  if item.is_multi_chat:
322
  selected_api_key = ASSISTANT_GOOGLE_API_KEYS or item.gemini_api_key
323
  oracle_base = ORACLE_TOKEN or item.oracle_base
 
331
  )
332
  cybersecuritydb(item.user_id, item.mongo_url)
333
  if item.oracle_base == "Delete":
334
+ clearedhistory = await geni._clear_oracle_history_in_db()
335
  return SuccessResponse(status="True", randydev={"message": f"Oracle Status: {clearedhistory}"})
336
  else:
337
+ answer, oracle_chat = await geni._GeminiLatest__get_response_oracle(item.query)
338
  return SuccessResponse(
339
  status="True",
340
  randydev={