add llamacloud config api

Author: leehuwuj

llama-index-server/llama_index/server/api/routers/chat.py

@@ -1,5 +1,6 @@
 import asyncio
 import logging
+import os
 from typing import AsyncGenerator, Callable, Union
 
 from fastapi import APIRouter, BackgroundTasks, HTTPException
@@ -15,6 +16,7 @@
 from llama_index.server.api.callbacks.stream_handler import StreamHandler
 from llama_index.server.api.models import ChatRequest
 from llama_index.server.api.utils.vercel_stream import VercelStreamResponse
+from llama_index.server.services.llamacloud import LlamaCloudFileService
 
 
 def chat_router(
@@ -57,6 +59,27 @@ async def chat(
             logger.error(e)
             raise HTTPException(status_code=500, detail=str(e))
 
+    if LlamaCloudFileService.is_configured():
+        @router.get("/config/llamacloud")
+        async def chat_llama_cloud_config():
+            if not os.getenv("LLAMA_CLOUD_API_KEY"):
+                raise HTTPException(
+                    status_code=500, detail="LlamaCloud API KEY is not configured"
+                )
+            projects = LlamaCloudFileService.get_all_projects_with_pipelines()
+            pipeline = os.getenv("LLAMA_CLOUD_INDEX_NAME")
+            project = os.getenv("LLAMA_CLOUD_PROJECT_NAME")
+            pipeline_config = None
+            if pipeline and project:
+                pipeline_config = {
+                    "pipeline": pipeline,
+                    "project": project,
+                }
+            return {
+                "projects": projects,
+                "pipeline": pipeline_config,
+            }
+
     return router
 
 

llama-index-server/llama_index/server/server.py

@@ -70,6 +70,7 @@ def _ui_config(self) -> dict:
         return {
             "CHAT_API": f"{self.api_prefix}/chat",
             "STARTER_QUESTIONS": self.starter_questions,
+            "LLAMA_CLOUD_API": "/api/chat/config/llamacloud",
         }
 
     # Default routers

llama-index-server/llama_index/server/services/llamacloud/file.py

@@ -8,11 +8,10 @@
 import requests
 from fastapi import BackgroundTasks
 from llama_cloud import ManagedIngestionStatus, PipelineFileCreateCustomMetadataValue
-from pydantic import BaseModel
-
 from llama_index.core.schema import NodeWithScore
 from llama_index.server.api.models import SourceNodes
 from llama_index.server.services.llamacloud.index import get_client
+from pydantic import BaseModel
 
 logger = logging.getLogger("uvicorn")
 
@@ -176,3 +175,10 @@ def _download_file(cls, url: str, local_file_path: str) -> None:
                 for chunk in r.iter_content(chunk_size=8192):
                     f.write(chunk)
         logger.info("File downloaded successfully")
+
+    @classmethod
+    def is_configured(cls) -> bool:
+        try:
+            return os.environ.get("LLAMA_CLOUD_API_KEY") is not None
+        except Exception:
+            return False