startup_apis / app /main.py
David Li
feat: basic app
deb8b74
raw
history blame contribute delete
706 Bytes
from fastapi import FastAPI
import os
from fastapi.responses import StreamingResponse, FileResponse
from io import BytesIO
from claude_api import Client
from fastapi.middleware.cors import CORSMiddleware
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/prompt/")
async def get_prompt(prompt: str = "Hello World"):
"""
"""
cookie = os.environ.get('CLAUDE_TOKEN', None)
claude_api = Client(cookie)
new_chat = claude_api.create_new_chat()
conversation_id = new_chat['uuid']
response = claude_api.send_message(prompt, conversation_id)
return all_files