github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-06 11:31:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

model server fix message model

Browse Source
This commit is contained in:
yhjun1026
2023-06-01 16:34:51 +08:00
parent 96c516ab55
commit 661a7b5697
11 changed files with 63 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/app.py
View File

@@ -3,7 +3,6 @@
import gradio as gr
from langchain.agents import AgentType, initialize_agent, load_tools
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
from llama_index import (
Document,
GPTSimpleVectorIndex,
@@ -12,7 +11,7 @@ from llama_index import (
ServiceContext,
)
from pilot.model.vicuna_llm import VicunaEmbeddingLLM, VicunaRequestLLM
from pilot.model.llm_out.vicuna_llm import VicunaEmbeddingLLM, VicunaRequestLLM
def agent_demo():
@@ -49,7 +48,7 @@ def get_answer(q):
def get_similar(q):
from pilot.vector_store.extract_tovec import knownledge_tovec, knownledge_tovec_st
from pilot.vector_store.extract_tovec import knownledge_tovec_st
docsearch = knownledge_tovec_st("./datasets/plan.md")
docs = docsearch.similarity_search_with_score(q, k=1)