From 4db666affd15a79d7e8f8cd1ba934b6ba6bf856a Mon Sep 17 00:00:00 2001 From: duwenxin99 Date: Thu, 21 Dec 2023 14:30:02 -0500 Subject: [PATCH] fix type --- langchain_tools_demo/main.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/langchain_tools_demo/main.py b/langchain_tools_demo/main.py index 64e1d230..d4e8ecc6 100644 --- a/langchain_tools_demo/main.py +++ b/langchain_tools_demo/main.py @@ -19,15 +19,16 @@ import uvicorn from fastapi import Body, FastAPI, HTTPException, Request -from fastapi.responses import HTMLResponse, PlainTextResponse, RedirectResponse +from fastapi.responses import PlainTextResponse, RedirectResponse from fastapi.staticfiles import StaticFiles from fastapi.templating import Jinja2Templates -from google.oauth2 import id_token from markdown import markdown from starlette.middleware.sessions import SessionMiddleware from agent import init_agent, user_agents +GOOGLE_REDIRECT_URI = "http://localhost:8081" + @asynccontextmanager async def lifespan(app: FastAPI): @@ -70,7 +71,7 @@ async def login_google( ): request.session.clear() form_data = await request.form() - user_id_token = form_data.get("credential", "") + user_id_token = str(form_data.get("credential", "")) # create new request session request.session["uuid"] = str(uuid.uuid4()) request.session["messages"] = BASE_HISTORY @@ -79,7 +80,11 @@ async def login_google( # Redirect to source URL source_url = request.headers.get("Referer") - return RedirectResponse(url=source_url) + source_url = request.headers.get("Referer") + if source_url: + return RedirectResponse(url=source_url) + else: + return RedirectResponse(url=GOOGLE_REDIRECT_URI) @app.post("/chat", response_class=PlainTextResponse)