jonathanjordan21 commited on
Commit
d75ac70
·
verified ·
1 Parent(s): 73b6736

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from fastapi import FastAPI
2
  import os
 
3
 
4
  from custom_llm import CustomLLM
5
 
@@ -7,8 +8,8 @@ from pydantic import BaseModel
7
  from langchain.prompts import PromptTemplate
8
 
9
  class ConversationPost(BaseModel):
10
- tenant: str | None = None
11
- module: str | None = None
12
  question: str
13
 
14
 
 
1
  from fastapi import FastAPI
2
  import os
3
+ from typing import Union
4
 
5
  from custom_llm import CustomLLM
6
 
 
8
  from langchain.prompts import PromptTemplate
9
 
10
  class ConversationPost(BaseModel):
11
+ tenant: Union[str, None] = None
12
+ module: Union[str, None] = None
13
  question: str
14
 
15