NickNYU commited on
Commit
1446765
1 Parent(s): fb94ec9

Delete langchain

Browse files
Files changed (2) hide show
  1. langchain/__init__.py +0 -0
  2. langchain/manager.py +0 -39
langchain/__init__.py DELETED
File without changes
langchain/manager.py DELETED
@@ -1,39 +0,0 @@
1
- from abc import abstractmethod, ABC
2
-
3
- from langchain.embeddings.base import Embeddings as LCEmbeddings
4
- from langchain.embeddings.openai import OpenAIEmbeddings
5
- from langchain.llms import AzureOpenAI
6
- from langchain.base_language import BaseLanguageModel
7
-
8
- from core.lifecycle import Lifecycle
9
-
10
-
11
- class BaseLangChainManager(Lifecycle, ABC):
12
- def __init__(self) -> None:
13
- super().__init__()
14
-
15
- @abstractmethod
16
- def get_embedding(self) -> LCEmbeddings:
17
- pass
18
-
19
- @abstractmethod
20
- def get_llm(self) -> BaseLanguageModel:
21
- pass
22
-
23
-
24
- class LangChainAzureManager(BaseLangChainManager):
25
- def __init__(self) -> None:
26
- super().__init__()
27
-
28
- # Override
29
- def get_embedding(self) -> LCEmbeddings:
30
- return OpenAIEmbeddings(client=None, chunk_size=1)
31
-
32
- # Override
33
- def get_llm(self) -> BaseLanguageModel:
34
- return AzureOpenAI(
35
- deployment_name="text-davinci-003",
36
- # model_name="text-davinci-003",
37
- model="text-davinci-003",
38
- client=None,
39
- )