Fixed OpenAI Calls

With the RunnableSequence we have to import AIMessage then also only after it is used we can strip the message.

app/main.py

@@ -19,6 +19,7 @@ from pydantic import BaseModel
 from langchain_openai import ChatOpenAI
 from langchain.prompts import PromptTemplate
 from langchain.schema.runnable import RunnableSequence
+from langchain.schema import AIMessage
 
 # Import the base LLM class to build our custom wrapper
 from langchain.llms.base import LLM
@@ -307,11 +308,27 @@ async def get_suggestions(
         )
         suggestion_chain: RunnableSequence = suggestion_prompt | llm
         # Run the chain without additional inputs, since prompt is fully baked
-        raw_text = await asyncio.to_thread(suggestion_chain.invoke, {})
+        raw_result = await asyncio.to_thread(suggestion_chain.invoke, {})
+
+        # 2) Normalize AIMessage or list[AIMessage] → plain string
+        if isinstance(raw_result, AIMessage):
+            raw_text = raw_result.content
+        elif (
+            isinstance(raw_result, (list, tuple))
+            and raw_result
+            and isinstance(raw_result[0], AIMessage)
+        ):
+            # If multiple AIMessage objects, join their contents
+            raw_text = "\n".join(msg.content for msg in raw_result)
+        else:
+            # Fallback to whatever __str__ gives you
+            raw_text = str(raw_result)
         # Strip out any leading numbers, bullets or whitespace
+        # 3) Now it’s safe to split on newlines and clean up bullets
         suggestion_array = [
             re.sub(r'^\s*[\-\d\.\)\s]+', '', line).strip()
-            for line in raw_text.split("\n") if line.strip()
+            for line in raw_text.split("\n")
+            if line.strip()
         ]
         if not suggestion_array:
             suggestion_array = ["No suggestions available"]

requirements.txt

@@ -1,12 +1,11 @@
 fastapi
 uvicorn
 httpx
-langchain[community]
+langchain
 langgraph
 openai
 pydantic
 loguru
 python-dotenv
 huggingface_hub
-langchain-community>=0.2.0
 langchain-openai