Update app/llm.py
Browse files- app/llm.py +19 -0
app/llm.py
CHANGED
@@ -11,6 +11,25 @@ from pydantic import BaseModel
|
|
11 |
from fastapi import APIRouter
|
12 |
from app.users import current_active_user
|
13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
14 |
class GenModel(BaseModel):
|
15 |
question: str
|
16 |
system: str = "You are a helpful medical AI chat assistant. Help as much as you can.Also continuously ask for possible symptoms in order to atat a conclusive ailment or sickness and possible solutions.Remember, response in English."
|
|
|
11 |
from fastapi import APIRouter
|
12 |
from app.users import current_active_user
|
13 |
|
14 |
+
from langchain_community.document_loaders import WebBaseLoader
|
15 |
+
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
16 |
+
from langchain_chroma import Chroma
|
17 |
+
from langchain_community.embeddings import GPT4AllEmbeddings
|
18 |
+
|
19 |
+
|
20 |
+
def agent():
|
21 |
+
loader = WebBaseLoader("https://lilianweng.github.io/posts/2023-06-23-agent/")
|
22 |
+
data = loader.load()
|
23 |
+
text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=0)
|
24 |
+
all_splits = text_splitter.split_documents(data)
|
25 |
+
|
26 |
+
|
27 |
+
def download_embedding():
|
28 |
+
vectorstore =Chroma.from_documents(documents=all_splits, embedding=GPT4AllEmbeddings())
|
29 |
+
|
30 |
+
|
31 |
+
|
32 |
+
|
33 |
class GenModel(BaseModel):
|
34 |
question: str
|
35 |
system: str = "You are a helpful medical AI chat assistant. Help as much as you can.Also continuously ask for possible symptoms in order to atat a conclusive ailment or sickness and possible solutions.Remember, response in English."
|