Create llms.py

llms.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+
+import os
+
+from dotenv import load_dotenv
+from langchain_google_genai import ChatGoogleGenerativeAI
+
+load_dotenv()
+
+
+def build_llm(model_name: str, temperature: float) -> ChatGoogleGenerativeAI:
+    api_key = os.getenv("GEMINI_API_KEY")
+    if not api_key:
+        raise ValueError("GEMINI_API_KEY is not set in the environment.")
+    return ChatGoogleGenerativeAI(
+        model=model_name,
+        google_api_key=api_key,
+        temperature=temperature,
+    )