Update endpoints.py

endpoints.py

@@ -1,8 +1,6 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from routers import dataset
-from routers import ocr
-from routers import chatgpt_plugin
+
 
 app = FastAPI(openapi_url="/api/v1/sparrow-data/openapi.json", docs_url="/api/v1/sparrow-data/docs")
 
@@ -14,11 +12,25 @@ app.add_middleware(
     allow_credentials=True,
 )
 
-app.include_router(dataset.router, prefix="/api-dataset/v1/sparrow-data", tags=["Dataset"])
-app.include_router(ocr.router, prefix="/api-ocr/v1/sparrow-data", tags=["OCR"])
-app.include_router(chatgpt_plugin.router, prefix="/api-chatgpt-plugin/v1/sparrow-data", tags=["ChatGPT Plugin"])
+
 
 
 @app.get("/")
 async def root():
-    return {"message": "R&D LLM API"}
+    return {"message": "R&D LLM API"}
+@app.get("/get")
+async def get():
+
+    return {"message": "try"}
+
+
+def askLLM(prompt):
+    output = query({
+        "inputs": prompt
+    })
+    return output
+
+@app.post("/ask_llm")
+async def ask_llm_endpoint(prompt: str):
+    result = askLLM(prompt)
+    return {"result": result}