Skip to main content

An academic paper writing agent based on LangGraph

Project description

seele-scholar-agent

English | 简体中文 | 日本語

PyPI version Python version CI License

LangGraph-based academic paper writing agent for topic proposal, literature retrieval, outline planning, section drafting, review loops, consistency checks, and reference generation.

Features

  • Search papers from OpenAlex, Semantic Scholar, ArXiv, and custom retrievers.
  • Propose concrete paper topics from broad research directions.
  • Generate paper-type-aware outlines with section purpose, transitions, evidence maps, and suggested figures or tables.
  • Draft sections with numbered citations, evidence packets, and claim-evidence bindings.
  • Review sections, revise drafts, and cap revisions per section without force-approving failed reviews.
  • Generate references with CrossRef/OpenAlex metadata enrichment and reject papers with no inline citations.
  • Check outline quality, citation validity, claim support, methodology, paragraph quality, terminology, logic, and citation consistency.
  • Apply locale-aware style packs for Chinese, Japanese, and English academic writing.
  • Stream node output with astream() for UI integration.

Install

git clone https://github.com/onekyuu/seele-scholar-agent
cd seele-scholar-agent

uv sync

For development dependencies:

uv sync --extra dev

Configuration

This package only owns agent-level configuration. LLM keys and models are injected by the caller.

Agent .env file:

cp src/seele_scholar_agent/.env.example src/seele_scholar_agent/.env

Supported agent variables:

Variable Description Default
SEMANTIC_SCHOLAR_API_KEY Optional Semantic Scholar API key empty
MAX_REVISIONS Max review cycles per section 3

Example caller environment:

OPENAI_API_KEY=sk-...
OPENAI_MODEL=gpt-4o-mini
OPENAI_BASE_URL=https://api.openai.com/v1

Any OpenAI-compatible endpoint can be used through ChatOpenAI.

Quality Controls

The workflow includes deterministic gates in addition to LLM review:

  • OutlineQualityGateNode: blocks incomplete outlines that lack purpose, transitions, target claims, evidence plans, or use an empirical template for a non-empirical paper type.
  • ReviewerNode: checks citation numbering, claim-source support, methodology/statistics issues, paragraph quality, and locale-specific writing style.
  • ReferenceGeneratorNode: returns NO_INLINE_CITATIONS instead of generating a full bibliography when the draft has no inline citations.
  • IntegrityGateNode: enforces strict academic checks when strict_academic_mode=True.
  • MaterialRegistry: always enforces citation boundaries for uploaded, external, background-only, excluded, trusted, normal, or low-confidence materials when provided.

The RAG context is represented as evidence packets with chunk_id, title, authors, year, page, section, relevance score, relevance rationale, and quote. Writer output is audited through ClaimEvidenceBinding so citations are checked against the claim they support, not only against reference numbers.

Caller State Options

The caller can pass optional fields in AgentState to control structure, evidence policy, and style:

state.update(
    {
        "paper_type": "literature_review",
        "structure_pattern": "thematic_review",
        "target_word_count": 6000,
        "strict_academic_mode": True,
        "writing_locale": "zh-CN",  # zh-CN, ja-JP, en-US, or a custom locale
        "style_profile": "thesis",
        "term_glossary": {"大语言模型": "大型语言模型"},
        "style_pack_override": {
            "display_name": "Custom thesis style",
            "general_guidance": ["Use the institution-specific academic style."],
        },
        "material_registry": {
            "entries": [
                {
                    "paper_id": "user-paper-1",
                    "source_origin": "user_upload",
                    "citation_role": "citable",
                    "confidence": "trusted",
                    "required": True,
                }
            ]
        },
        "check_required_material_relevance": True,
    }
)

material_registry source-boundary checks are always active when a registry is provided. The required-material relevance check is optional and only runs when check_required_material_relevance=True. Required citations are marked with material_registry.entries[].required=True.

Quick Start

Run the no-interrupt workflow:

export OPENAI_API_KEY="sk-..."
export SCHOLAR_TOPIC="large language model interpretability"
export SCHOLAR_LANGUAGE="zh"

uv run python examples/simple_workflow.py

Use the graph in your own code:

from langchain_openai import ChatOpenAI

from seele_scholar_agent.graph import create_simple_writing_graph
from examples.common import build_initial_state, build_prompts

model = ChatOpenAI(model="gpt-4o-mini", api_key="sk-...")
state = build_initial_state("large language model interpretability")

app = create_simple_writing_graph(
    model=model,
    prompts=build_prompts(),
    rag_retriever=None,
)

result = await app.ainvoke(
    state,
    config={"configurable": {"thread_id": state["thread_id"]}},
)

Examples

Large runnable examples live in examples/:

File Purpose
examples/common.py Shared model, prompt, and initial-state helpers
examples/simple_workflow.py Full automatic workflow with create_simple_writing_graph
examples/full_workflow_with_interrupts.py Human-in-the-loop topic and outline approval
examples/custom_retrievers.py Inject custom RAG and paper retrievers
examples/stream_nodes.py Stream a single node with astream()
examples/figure_placeholders.py Parse {{FIGURE: ...}} and {{TABLE: ...}} placeholders

Run any example from the repository root:

uv run python examples/full_workflow_with_interrupts.py

Core API

The package exposes two graph builders:

from seele_scholar_agent.graph import create_simple_writing_graph, create_writing_graph
  • create_writing_graph(...): includes interrupts after topic proposal and outline planning.
  • create_simple_writing_graph(...): runs without interrupts and is useful for tests or batch jobs.

Both require:

  • model: a ChatOpenAI instance or compatible LangChain chat model.
  • prompts: a complete PromptsConfig.
  • rag_retriever: optional Callable[[str], Awaitable[list[DocumentChunk]]].

Optional paper sources can be injected with extra_paper_retrievers.

Workflow

START
  -> topic_proposer
  -> researcher
  -> planner
  -> outline_quality_gate
  -> writer <-> reviewer
  -> finalizer
  -> reference_generator
  -> consistency_checker
  -> integrity_gate
  -> END

create_writing_graph() interrupts after topic_proposer and planner so the caller can choose a topic and approve the outline.

Project Structure

src/seele_scholar_agent/
├── agent_config.py
├── config.py
├── graph.py
├── i18n.py
├── logging.py
├── state.py
├── style_packs.py
├── nodes/
│   ├── topic_proposer.py
│   ├── researcher.py
│   ├── planner.py
│   ├── outline_quality_gate.py
│   ├── writer.py
│   ├── reviewer.py
│   ├── finalizer.py
│   ├── reference_generator.py
│   ├── consistency_checker.py
│   ├── integrity_gate.py
│   └── language_style_audit.py
└── tools/
    └── crossref.py

Development

uv run pytest
uv run ruff check src/
uv run mypy src/

Build the package:

uv build

License

MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

seele_scholar_agent-0.12.0.tar.gz (280.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

seele_scholar_agent-0.12.0-py3-none-any.whl (68.9 kB view details)

Uploaded Python 3

File details

Details for the file seele_scholar_agent-0.12.0.tar.gz.

File metadata

  • Download URL: seele_scholar_agent-0.12.0.tar.gz
  • Upload date:
  • Size: 280.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.9.17 {"installer":{"name":"uv","version":"0.9.17","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for seele_scholar_agent-0.12.0.tar.gz
Algorithm Hash digest
SHA256 3f2a464c831d519bd12a148075d0ea9f2ce9bbfa3efd1f4853dd9779ec67e706
MD5 10df5e96f9f5b86345f97538dab5a915
BLAKE2b-256 eae323c1f96aa257a2be6658b0f1b47271ed413c610e744050304875453821dc

See more details on using hashes here.

File details

Details for the file seele_scholar_agent-0.12.0-py3-none-any.whl.

File metadata

  • Download URL: seele_scholar_agent-0.12.0-py3-none-any.whl
  • Upload date:
  • Size: 68.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.9.17 {"installer":{"name":"uv","version":"0.9.17","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for seele_scholar_agent-0.12.0-py3-none-any.whl
Algorithm Hash digest
SHA256 b755938573a9aa83e701202196d480262c9f9d816deab697cbc841e483b25f7b
MD5 a831e0a4fe16a998d7f3a755eb64c86e
BLAKE2b-256 ea584f334d13987eb6c167c0ae0bed30561fc2e1645decc6cccb359a35d35db7

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page