testing rag

app.py

@@ -15,6 +15,7 @@ from langchain.text_splitter import CharacterTextSplitter
 from azure.core.credentials import AzureKeyCredential
 from azure.ai.inference import EmbeddingsClient
 import cassio
+from pydantic import BaseModel
 import shutil
 from config import settings
 app = FastAPI()
@@ -49,6 +50,8 @@ llm = None
 embedding = None
 cassio.init(token=ASTRA_DB_APPLICATION_TOKEN, database_id=ASTRA_DB_ID)
 
+class MessageRequest(BaseModel):
+    userMessage: str
 class AzureOpenAIEmbeddings:
     def __init__(self, client):
         self.client = client