Update functions.py

functions.py

@@ -1,6 +1,50 @@
-from some_llm_library import PromptTemplate, StrOutputParser
-
-
+import os
+from typing_extensions import TypedDict, List
+from IPython.display import Image, display
+from langchain_core.pydantic_v1 import BaseModel, Field
+from langchain.schema import Document
+from langgraph.graph import START, END, StateGraph
+from langchain.prompts import PromptTemplate
+import uuid
+from langchain_exa import ExaSearchRetriever, TextContentsOptions
+from langchain_groq import ChatGroq
+from langchain_community.utilities import GoogleSerperAPIWrapper
+from langchain_chroma import Chroma
+from langchain_community.document_loaders import NewsURLLoader
+from langchain_community.retrievers.wikipedia import WikipediaRetriever
+from sentence_transformers import SentenceTransformer
+from langchain.vectorstores import Chroma
+from langchain_community.document_loaders import UnstructuredURLLoader, NewsURLLoader
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain_community.document_loaders import WebBaseLoader
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.output_parsers import JsonOutputParser
+from langchain_community.vectorstores.utils import filter_complex_metadata
+from langchain.schema import Document
+from langgraph.graph import START, END, StateGraph
+
+
+
+async def predict_custom_agent_answer(example: dict):
+    config = {"configurable": {"thread_id": str(uuid.uuid4())}}
+    
+    try:
+        # Invoke the custom graph with the input question
+        state_dict = await  custom_graph.ainvoke(
+            {"question": example["input"], "steps": []}, config
+        )
+        
+        # Check if 'generation' exists and is not empty
+        if 'generation' in state_dict and state_dict['generation']:
+            return {"response": state_dict["generation"], "steps": state_dict["steps"]}
+        else:
+            # Raise an exception if the generation is missing or empty
+            print("Your question violates toxicity rules or contains sensitive information.")
+    
+    except Exception as e:
+        # Raise the exception with a custom message
+        print("An error occurred: Try to change the question.")