File size: 848 Bytes
3505899 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from langchain.chat_models import ChatOpenAI
from chains.output_format.base import chain_output_format
from chains.code_generator.base import chain_code_generator
import os
class GPTAssistant():
def __init__(self,api_key:str):
os.environ['OPENAI_API_KEY'] = api_key
self.llm = ChatOpenAI(temperature=0, model_name='gpt-3.5-turbo', request_timeout=120, client=None)
def chain_response_format(self, html_content):
# prompt templates
output_format_chain = chain_output_format(self.llm)
# chain
return output_format_chain.run(html_content=html_content)
def chain_code_generator(self, output_format, html_content):
# Prompt templates
script_chain = chain_code_generator(self.llm)
return script_chain.run(output_format=output_format, html_content=html_content)
|