Damien Benveniste commited on
Commit
5eea5ad
·
1 Parent(s): 54e091e
Files changed (2) hide show
  1. app.py +17 -2
  2. requirements.txt +4 -1
app.py CHANGED
@@ -1,7 +1,22 @@
1
- from fastapi import FastAPI
 
 
2
 
3
  app = FastAPI()
4
 
5
  @app.get("/")
6
  def greet_json():
7
- return {"Hello": "World!"}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI, Request, Response
2
+ from langserve import APIHandler
3
+ from langchain_huggingface import HuggingFaceEndpoint
4
 
5
  app = FastAPI()
6
 
7
  @app.get("/")
8
  def greet_json():
9
+ return {"Hello": "World!"}
10
+
11
+
12
+
13
+ llm = HuggingFaceEndpoint(
14
+ repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
15
+ )
16
+
17
+ api_handler = APIHandler(llm)
18
+
19
+
20
+ @app.get("/stream")
21
+ async def simple_stream(request: Request):
22
+ return await api_handler.stream(request)
requirements.txt CHANGED
@@ -1,2 +1,5 @@
1
  fastapi
2
- uvicorn[standard]
 
 
 
 
1
  fastapi
2
+ uvicorn[standard]
3
+ langserve
4
+ langchain
5
+ langchain-huggingface