Update main.py

main.py

@@ -16,16 +16,7 @@ app.add_middleware(  # add the middleware
 def hello():
     return "Hello, I'm Artist"
 
-@app.post('/generate_completion')
-async def generate_completion(
-    model: str = Query('gpt-4', description='The model to use for generating the completion'),
-    messages: List[Dict[str, str]] = Query(..., description='The list of messages to generate the completion for'),
-    stream: bool = Query(False, description='Whether to stream the response')
-):
-    response = index._create_completion(model=model, messages=messages, stream=stream)
-    
-    result = []
-    for message in response:
-        result.append(message)
-    
-    return result
+@app.post("/generate")
+def generate_completion(message: Message):
+    response = Linlada._create_completion(model='gpt-4', messages=[{"role": message.role, "content": message.content}], stream=False)
+    return response