File size: 612 Bytes
194e0e0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import os
import google.generativeai as genai
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_google_genai import GoogleGenerativeAIEmbeddings
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_google_genai import GoogleGenerativeAIEmbeddings
#lấy API gemini
os.environ["GOOGLE_API_KEY"] = ''
genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
#lấy model chatbot
model = ChatGoogleGenerativeAI(model="gemini-pro",
temperature=0.4)
#lấy model embedding
embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001") |