Skip to content

Commit

Permalink
Fix python code format.
Browse files Browse the repository at this point in the history
  • Loading branch information
shaoyijia committed Sep 25, 2024
1 parent ded466d commit efac123
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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
"""
Expand Down Expand Up @@ -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.
"""

Expand Down
1 change: 1 addition & 0 deletions knowledge_storm/rm.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit efac123

Please sign in to comment.