highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-18 12:00:58 +02:00
Code Issues Packages Projects Releases Wiki Activity

Custom Model Server Note (#668)

Browse Source
This commit is contained in:
Yuhong Sun 2023-10-31 19:05:38 -07:00 committed by GitHub
parent fbf7c642a3
commit 53353f9b62
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/danswer/llm/custom_llm.py
View File

@ -15,7 +15,11 @@ class CustomModelServer(LLM):
"""This class is to provide an example for how to use Danswer """This class is to provide an example for how to use Danswer
with any LLM, even servers with custom API definitions. with any LLM, even servers with custom API definitions.
To use with your own model server, simply implement the functions To use with your own model server, simply implement the functions
below to fit your model server expectation""" below to fit your model server expectation
The implementation below works against the custom FastAPI server from the blog:
https://medium.com/@yuhongsun96/how-to-augment-llms-with-private-data-29349bd8ae9f
"""
def __init__( def __init__(
self, self,