moriire commited on
Commit
97da889
1 Parent(s): 73fe91e

Update app/llm.py

Browse files
Files changed (1) hide show
  1. 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."