Spaces:
Runtime error
Runtime error
File size: 638 Bytes
5206fbb 010c22e 5206fbb a75f10f 5206fbb |
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 |
"""Module for building the Langchain Agent"""
import streamlit as st
from langchain_openai import ChatOpenAI
from gnosis.agent import PDFExplainer
def build(key, client):
"""An Agent builder"""
# Build Agent
try:
llm = ChatOpenAI(
temperature=st.session_state.temperature,
model="gpt-4o-mini",
api_key=key,
)
agent = PDFExplainer(
llm,
client,
extra_tools=st.session_state.wk_button,
).agent
except Exception: # pylint: disable=broad-exception-caught
st.warning("Missing OpenAI API Key.")
return agent
|