Spaces:
Sleeping
Sleeping
File size: 657 Bytes
3e595a5 95243ee 3e595a5 95243ee 3e595a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import os
from dotenv import load_dotenv
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_openai import ChatOpenAI
from langchain_ollama import ChatOllama
def get_llm():
# Load environment variables from .env file
load_dotenv()
# Get the API key
# google_api_key = os.getenv("GOOGLE_API_KEY")
# llm = ChatOpenAI(temperature=0)
# llm = ChatOllama(model="llama3.2:latest", temperature=0)
"""llm = ChatOpenAI(
api_key="ollama",
model="llama3.2:latest",
base_url="http://141.211.127.171/",
)"""
llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash-latest")
return llm |