+
Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions lib/sycamore/sycamore/docset.py
Original file line number Diff line number Diff line change
Expand Up @@ -1140,7 +1140,6 @@ def rerank(
query: str,
score_property_name: str = "_rerank_score",
limit: Optional[int] = None,
**kwargs,
) -> "DocSet":
"""
Sort a DocSet given a scoring class.
Expand All @@ -1158,7 +1157,7 @@ def rerank(
else:
plan = self.plan
similarity_scored = ScoreSimilarity(
plan, similarity_scorer=similarity_scorer, query=query, score_property_name=score_property_name, **kwargs
plan, similarity_scorer=similarity_scorer, query=query, score_property_name=score_property_name
)
return DocSet(
self.context,
Expand Down
2 changes: 1 addition & 1 deletion lib/sycamore/sycamore/query/execution/sycamore_executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,7 @@ def process_node(self, logical_node: Node, query_id: str, is_result_node: Option
else:
raise ValueError(f"Unsupported node type: {str(logical_node)}")

code, imports = operation.script(output_var=(self.OUTPUT_VAR_NAME if not is_result_node else None))
code, imports = operation.script(output_var=(self.OUTPUT_VAR_NAME if is_result_node else None))
self.imports += imports
self.node_id_to_code[logical_node.node_id] = code
self.node_id_to_node[logical_node.node_id] = logical_node
Expand Down
7 changes: 3 additions & 4 deletions lib/sycamore/sycamore/tests/unit/test_docset.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
from sycamore import DocSet, Context
from sycamore.context import OperationTypes, ExecMode
from sycamore.data import Document, Element
from sycamore.llms import LLM
from sycamore.llms.prompts.default_prompts import (
LlmClusterEntityAssignGroupsMessagesPrompt,
LlmClusterEntityFormGroupsMessagesPrompt,
Expand All @@ -28,16 +29,14 @@
ExtractProperties,
Query,
)

from sycamore.llms import LLM
from sycamore.transforms import Filter
from sycamore.transforms.base import get_name_from_callable
from sycamore.transforms.extract_entity import OpenAIEntityExtractor
from sycamore.transforms.extract_schema import SchemaExtractor
from sycamore.transforms import Filter
from sycamore.transforms.query import QueryExecutor
from sycamore.transforms.similarity import SimilarityScorer
from sycamore.transforms.sort import Sort
from sycamore.transforms.summarize import LLMElementTextSummarizer
from sycamore.transforms.query import QueryExecutor


class MockLLM(LLM):
Expand Down
Loading
点击 这是indexloc提供的php浏览器服务,不要输入任何密码和下载