Update main.py
Browse files
main.py
CHANGED
@@ -62,18 +62,18 @@ def gpt_with_google_search(prompt):
|
|
62 |
|
63 |
class Linlada(BaseModel):
|
64 |
prompt: str
|
65 |
-
web_access: str
|
66 |
-
|
67 |
-
|
68 |
-
neg: str = None
|
69 |
|
70 |
|
71 |
@app.post('/linlada')
|
72 |
def linlada(request: Linlada):
|
73 |
-
llm = G4F(model=model)
|
74 |
prompt = request.prompt
|
|
|
75 |
web_access = request.web_access
|
76 |
-
|
|
|
77 |
chat = gpt_with_google_search(prompt)
|
78 |
else:
|
79 |
chat = llm(prompt)
|
|
|
62 |
|
63 |
class Linlada(BaseModel):
|
64 |
prompt: str
|
65 |
+
web_access: str
|
66 |
+
model: str = 'gpt-3.5-turbo'
|
67 |
+
|
|
|
68 |
|
69 |
|
70 |
@app.post('/linlada')
|
71 |
def linlada(request: Linlada):
|
|
|
72 |
prompt = request.prompt
|
73 |
+
model = request.model
|
74 |
web_access = request.web_access
|
75 |
+
llm = G4F(model=model)
|
76 |
+
if web_access == "true":
|
77 |
chat = gpt_with_google_search(prompt)
|
78 |
else:
|
79 |
chat = llm(prompt)
|