Update pipeline.py

pipeline.py

@@ -68,16 +68,16 @@ def classify_query(query: str) -> str:
     return classification if classification != "OutOfScope" else "OutOfScope"
 
 # Function to moderate text using Mistral moderation API (async version)
-async def moderate_text(query: str) -> str:
+def moderate_text(query: str) -> str:
     try:
         # Use Pydantic AI to validate the text
-        await pydantic_agent.run(query)  # Use async run for Pydantic validation
+        pydantic_agent.run_sync(query)  # Use sync run for Pydantic validation
     except Exception as e:
         print(f"Error validating text: {e}")
         return "Invalid text format."
     
     # Call the Mistral moderation API
-    response = await client.classifiers.moderate_chat(
+    response = client.classifiers.moderate_chat(
         model="mistral-moderation-latest",
         inputs=[{"role": "user", "content": query}]
     )
@@ -201,7 +201,8 @@ async def run_async_pipeline(query: str) -> str:
 
 # Run the pipeline with the event loop
 def run_with_chain(query: str) -> str:
-    return asyncio.run(run_async_pipeline(query))
+    loop = asyncio.get_event_loop()
+    return loop.run_until_complete(run_async_pipeline(query))
 
 # Initialize chains here
 classification_chain = get_classification_chain()