Spaces:
Sleeping
Sleeping
File size: 2,912 Bytes
0743bb0 9002555 0743bb0 9002555 0743bb0 9002555 0743bb0 9002555 0743bb0 9002555 0743bb0 9002555 0743bb0 9002555 0743bb0 9002555 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
from fastapi import APIRouter, HTTPException, Depends
from service.dto import UserPromptRequest, BotResponse
from core.chat.chatstore import ChatStore
from api.function import (
generate_streaming_completion,
generate_completion_non_streaming,
)
from sse_starlette.sse import EventSourceResponse
from utils.utils import generate_uuid
router = APIRouter(tags=["Bot"])
def get_chat_store():
return ChatStore()
@router.post("/bot/new")
async def create_new_session():
session_id = generate_uuid()
return {"session_id" : session_id}
@router.get("/bot/{session_id}")
async def get_session_id(session_id: str, chat_store: ChatStore = Depends(get_chat_store)):
chat_history = chat_store.get_messages(session_id)
if not chat_history:
raise HTTPException(status_code=404, detail="Session not found or empty.")
return chat_history
@router.get("/bot")
async def get_all_session_ids():
try:
chat_store = ChatStore()
all_keys = chat_store.get_keys()
print(all_keys)
return all_keys
except Exception as e:
# Log the error and raise HTTPException for FastAPI
print(f"An error occurred in update data.: {e}")
raise HTTPException(
status_code=400, detail="the error when get all session ids"
)
@router.post("/bot/{session_id}")
async def bot_generator_general(user_prompt_request: UserPromptRequest):
if user_prompt_request.streaming:
return EventSourceResponse(
generate_streaming_completion(
user_prompt_request.prompt, user_prompt_request.streaming
)
)
else:
response, raw_references, references, metadata, scores = (
generate_completion_non_streaming(
user_prompt_request.session_id, user_prompt_request.prompt, user_prompt_request.streaming
)
)
return BotResponse(
content=response,
raw_references=raw_references,
references=references,
metadata=metadata,
scores=scores,
)
@router.post("/bot/{category_id}/{title}") #Ganti router
async def bot_generator_spesific(
category_id: int, title: str, user_prompt_request: UserPromptRequest
):
pass
@router.delete("/bot/{session_id}")
async def delete_bot(session_id: str, chat_store: ChatStore = Depends(get_chat_store)):
try:
chat_store.delete_messages(session_id)
return {"info": f"Delete {session_id} successful"}
except Exception as e:
# Log the error and raise HTTPException for FastAPI
print(f"An error occurred in update data.: {e}")
raise HTTPException(
status_code=400, detail="the error when deleting message"
)
@router.get("/bot/{category_id}/{title}")
async def get_favourite_data(category_id: int, title: str, human_template):
pass |