Update app.py

app.py

@@ -343,6 +343,8 @@ class DocumentRAG:
         return {"retrieved_docs": docs, "question": query}
 
 
+
+
     def rerank_node(self, state: StoryState):
         topic = state.story_topic
         query = f"Rerank documents based on how well they explain the topic {topic}"
@@ -352,7 +354,7 @@ class DocumentRAG:
         if not texts:
             return {"reranked_docs": [], "question": query}
 
-        if self.embedding_choice == "Cohere":
+        if self.embedding_choice == "Cohere" and hasattr(self, "cohere_client"):
             rerank_results = self.cohere_client.rerank(
                 query=query,
                 documents=texts,
@@ -368,6 +370,7 @@ class DocumentRAG:
 
 
 
+
     def generate_story_node(self, state: StoryState):
         context = "\n\n".join(state.reranked_docs)
         topic = state.story_topic