feat: update tests and streamlit_app

aiagents4pharma/talk2cells/agents/agent_scp.py

@@ -5,13 +5,11 @@
 '''
 
 import os
-import sys
 from langchain_openai import ChatOpenAI
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.graph import START, StateGraph
-from langgraph.prebuilt import ToolNode
+# from langgraph.prebuilt import ToolNode
 from langgraph.prebuilt import create_react_agent
-sys.path.append('..')
 from ..tools.search_studies import search_studies
 from ..tools.display_studies import display_studies
 from ..states.state_talk2cells import Talk2Cells
@@ -44,7 +42,7 @@ def agent_scp_node(state: Talk2Cells):
     # Create a ToolNode
     # This node will call the tools based
     # on the response from the LLM
-    tool_node = ToolNode(tools)
+    # tool_node = ToolNode(tools)
 
     # Create the LLM
     # And bind the tools to it

aiagents4pharma/talk2cells/tests/test_scp_agent.py

@@ -0,0 +1,23 @@
+'''
+Test cases for the search_studies
+'''
+
+# from ..tools.search_studies import search_studies
+from aiagents4pharma.talk2cells.agents.agent_scp import get_app
+from langchain_core.messages import HumanMessage
+
+def test_agent_scp():
+    '''
+    Test the agent_scp.
+    '''
+    unique_id = 12345
+    app = get_app(unique_id)
+    config = {"configurable": {"thread_id": unique_id}}
+    prompt = "Search for studies on Crohns Disease."
+    response = app.invoke(
+                        {"messages": [HumanMessage(content=prompt)]},
+                        config=config
+                    )
+    assistant_msg = response["messages"][-1].content
+    # Check if the assistant message is a string
+    assert isinstance(assistant_msg, str)

aiagents4pharma/talk2cells/tools/search_studies.py

@@ -42,7 +42,7 @@ def search_studies(search_term: str,
     # with the selected columns
     df = pd.DataFrame(search_response.json()['studies'])[selected_columns]
 
-    # Convert column 'Study Name' into clickable 
+    # Convert column 'Study Name' into clickable
     # hyperlinks from the column 'Study URL'
     scp_api_url = 'https://singlecell.broadinstitute.org'
     df['name'] = df.apply(

aiagents4pharma/talk2cells/streamlit_app.py → app/frontend/streamlit_app_talk2cells.py

@@ -4,14 +4,15 @@
 Talk2Cells: A Streamlit app for the Talk2Cells graph.
 '''
 
-from langchain_core.messages import HumanMessage
-from agents.agent_scp import get_app
 import os
-import streamlit as st
+import sys
 import random
+import streamlit as st
 from langchain_core.messages import SystemMessage, HumanMessage, AIMessage
 from langchain_core.messages import ChatMessage
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+sys.path.append('./')
+from aiagents4pharma.talk2cells.agents.agent_scp import get_app
 
 st.set_page_config(page_title="Talk2Cells", page_icon="🤖", layout="wide")