Skip to content

Commit

Permalink
Update dependency and move to gpt-4-turbo
Browse files Browse the repository at this point in the history
  • Loading branch information
VVoruganti committed Dec 12, 2023
1 parent fe39923 commit 9409bbc
Show file tree
Hide file tree
Showing 4 changed files with 870 additions and 802 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -158,3 +158,5 @@ cython_debug/
# and can be added to the global gitignore or merged into this file. For a more nuclear
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
#.idea/

.DS_Store
6 changes: 2 additions & 4 deletions mediator.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from langchain.schema.messages import BaseMessage, _message_to_dict, messages_from_dict
from langchain.vectorstores import SupabaseVectorStore
from langchain.embeddings.base import Embeddings
from langchain.embeddings.openai import OpenAIEmbeddings
from langchain.embeddings import AzureOpenAIEmbeddings
import uuid
import sentry_sdk
import os
Expand All @@ -24,11 +24,9 @@ def __init__(self):
self.conversation_table = os.environ["CONVERSATION_TABLE"]
self.match_function = os.environ["MATCH_FUNCTION"]

embeddings = OpenAIEmbeddings(
embeddings = AzureOpenAIEmbeddings(
deployment=os.environ["OPENAI_API_EMBEDDING_NAME"],
model="text-embedding-ada-002",
openai_api_base=os.environ["OPENAI_API_BASE"],
openai_api_type=os.environ["OPENAI_API_TYPE"],
)
self.vector_table = SupabaseVectorStore(
embedding=embeddings,
Expand Down
Loading

0 comments on commit 9409bbc

Please sign in to comment.