Skip to content

Multi paper recommendation

This tool is used to return recommendations based on multiple papers

MultiPaperRecInput

Bases: BaseModel

Input schema for multiple paper recommendations tool.

Source code in aiagents4pharma/talk2scholars/tools/s2/multi_paper_rec.py 
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
class MultiPaperRecInput(BaseModel):
    """Input schema for multiple paper recommendations tool."""

    paper_ids: List[str] = Field(
        description="List of Semantic Scholar Paper IDs to get recommendations for"
    )
    limit: int = Field(
        default=10,
        description="Maximum total number of recommendations to return",
        ge=1,
        le=500,
    )
    year: Optional[str] = Field(
        default=None,
        description="Year range in format: YYYY for specific year, "
        "YYYY- for papers after year, -YYYY for papers before year, or YYYY:YYYY for range",
    )
    tool_call_id: Annotated[str, InjectedToolCallId]

    model_config = {"arbitrary_types_allowed": True}

get_multi_paper_recommendations(paper_ids, tool_call_id, limit=2, year=None)

Get recommendations for a group of multiple papers using the Semantic Scholar IDs. No other paper IDs are supported.

Parameters:

Name Type Description Default
paper_ids List[str]

The list of paper IDs to base recommendations on.

 required
tool_call_id Annotated[str, InjectedToolCallId]

The tool call ID.

 required
limit int

The maximum number of recommendations to return. Defaults to 2.

 2
year str

Year range for papers.

 None
Supports formats like "2024-", "-2024", "2024

2025". Defaults to None.

 required

Returns:

Type Description
Command[Any]

Dict[str, Any]: The recommendations and related information.
Source code in aiagents4pharma/talk2scholars/tools/s2/multi_paper_rec.py 
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
@tool(args_schema=MultiPaperRecInput, parse_docstring=True)
def get_multi_paper_recommendations(
    paper_ids: List[str],
    tool_call_id: Annotated[str, InjectedToolCallId],
    limit: int = 2,
    year: Optional[str] = None,
) -> Command[Any]:
    """
    Get recommendations for a group of multiple papers using the Semantic Scholar IDs.
    No other paper IDs are supported.

    Args:
        paper_ids (List[str]): The list of paper IDs to base recommendations on.
        tool_call_id (Annotated[str, InjectedToolCallId]): The tool call ID.
        limit (int, optional): The maximum number of recommendations to return. Defaults to 2.
        year (str, optional): Year range for papers.
        Supports formats like "2024-", "-2024", "2024:2025". Defaults to None.

    Returns:
        Dict[str, Any]: The recommendations and related information.
    """
    # Create recommendation data object to organize variables
    rec_data = MultiPaperRecData(paper_ids, limit, year, tool_call_id)

    # Process the recommendations
    results = rec_data.process_recommendations()

    return Command(
        update={
            "multi_papers": results["papers"],
            "last_displayed_papers": "multi_papers",
            "messages": [
                ToolMessage(
                    content=results["content"],
                    tool_call_id=tool_call_id,
                    artifact=results["papers"],
                )
            ],
        }
    )