52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
import math
|
|
import numpy
|
|
from dotenv import load_dotenv
|
|
from bs4 import BeautifulSoup
|
|
from nltk.tokenize import WordPunctTokenizer, RegexpTokenizer
|
|
from sklearn.metrics.pairwise import cosine_similarity
|
|
|
|
from langchain import hub
|
|
from langchain.chains import LLMChain
|
|
from langchain.memory import ConversationBufferMemory
|
|
from langchain.prompts import (
|
|
MessagesPlaceholder,
|
|
HumanMessagePromptTemplate,
|
|
ChatPromptTemplate,
|
|
PromptTemplate,
|
|
)
|
|
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
|
from langchain_core.messages import HumanMessage, SystemMessage
|
|
from langchain_core.output_parsers import StrOutputParser
|
|
from langchain_core.runnables import RunnablePassthrough
|
|
from langchain_google_genai import (
|
|
GoogleGenerativeAI,
|
|
GoogleGenerativeAIEmbeddings,
|
|
ChatGoogleGenerativeAI,
|
|
HarmCategory,
|
|
HarmBlockThreshold,
|
|
)
|
|
from langchain_community.document_loaders import AsyncHtmlLoader, RecursiveUrlLoader
|
|
from langchain_community.document_transformers import BeautifulSoupTransformer
|
|
from langchain_community.vectorstores import Chroma
|
|
|
|
from langchain_openai import ChatOpenAI
|
|
from langchain_openai import OpenAI
|
|
|
|
from langchain_core.messages import HumanMessage
|
|
|
|
load_dotenv()
|
|
llm = OpenAI()
|
|
chat_model = ChatOpenAI(model="gpt-4")
|
|
|
|
text = "What is a good question to put here?"
|
|
messages = [HumanMessage(content=text)]
|
|
|
|
llm.invoke(text)
|
|
# >> Feetful of Fun
|
|
|
|
chat_model.invoke(messages)
|
|
# >> AIMessage(content="Socks O'Color")
|