From efac123871781392e2c2bca24670cd9aba9588c7 Mon Sep 17 00:00:00 2001 From: shaoyijia Date: Wed, 25 Sep 2024 10:22:38 -0700 Subject: [PATCH] Fix python code format. --- .../collaborative_storm/modules/co_storm_agents.py | 4 ++-- knowledge_storm/rm.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/knowledge_storm/collaborative_storm/modules/co_storm_agents.py b/knowledge_storm/collaborative_storm/modules/co_storm_agents.py index e092cd6..e7f6029 100644 --- a/knowledge_storm/collaborative_storm/modules/co_storm_agents.py +++ b/knowledge_storm/collaborative_storm/modules/co_storm_agents.py @@ -110,7 +110,7 @@ def generate_utterance( class SimulatedUser(Agent): """ Simulated Users is a special type of Agent in Co-STORM that simulates real user interaction behavior based on the given intent. - + This class can be used for automatic experiments. For more information, please refer to Section 3.4 of Co-STORM paper: https://www.arxiv.org/pdf/2408.15232 """ @@ -321,7 +321,7 @@ class PureRAGAgent(Agent): """ PureRAGAgent only handles grounded question generation by retrieving information from the retriever based on the query. It does not utilize any other information besides the query itself. - + It's designed for Co-STORM paper baseline comparison. """ diff --git a/knowledge_storm/rm.py b/knowledge_storm/rm.py index 9ae5955..7f029e7 100644 --- a/knowledge_storm/rm.py +++ b/knowledge_storm/rm.py @@ -335,6 +335,7 @@ def forward(self, query_or_queries: Union[str, List[str]], exclude_urls: List[st class StanfordOvalArxivRM(dspy.Retrieve): """[Alpha] This retrieval class is for internal use only, not intended for the public.""" + def __init__(self, endpoint, k=3): super().__init__(k=k) self.endpoint = endpoint