Refactor MyPythonClass build method to accept openai_api_key parameter.

- The `build` method of `MyPythonClass` is modified to accept an `openai_api_key` parameter instead of using a hardcoded value.
- This change allows for dynamic usage of different API keys when constructing an instance of `ConversationChain`.
- The `my_conversation` method now takes `openai_api_key` as an argument and uses it to initialize the `llm` instance of `OpenAI`.
- The `build` method simply delegates the `openai_api_key` argument to the `my_conversation` method and returns the resulting `ConversationChain` instance.
This commit is contained in:
gustavoschaedler 2023-07-05 03:37:18 +01:00
commit 48345dd8dd

View file

@ -56,19 +56,17 @@ from langchain.memory import ConversationBufferMemory
class MyPythonClass:
def my_conversation(self):
def my_conversation(self, openai_api_key):
llm = OpenAI(
openai_api_key='',
openai_api_key=openai_api_key,
temperature=0
)
return ConversationChain(
llm=llm, verbose=True, memory=ConversationBufferMemory()
)
def build(self, my_custom_input: str) -> ConversationChain:
conversation = self.my_conversation()
return conversation
def build(self, openai_api_key: str) -> ConversationChain:
return self.my_conversation(openai_api_key)
"""
DIRECT_TYPES = ["str", "bool", "code", "int", "float", "Any", "prompt"]