wehub-resource-sync
e4dcfc49aa
Tests / Import Check (Python 3.13) (push) Has been cancelled
Tests / Import Check (Python 3.14) (push) Has been cancelled
Tests / Python Tests (Python 3.11) (push) Has been cancelled
Tests / Python Tests (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.14) (push) Has been cancelled
Tests / Test Summary (push) Has been cancelled
Tests / Lint and Format (push) Has been cancelled
Tests / Web Node Tests (push) Has been cancelled
Tests / Import Check (Python 3.11) (push) Has been cancelled
Tests / Import Check (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.13) (push) Has been cancelled
105 lines
3.0 KiB
Python
105 lines
3.0 KiB
Python
"""
|
|
Brainstorm tool - stateless breadth-first idea exploration.
|
|
|
|
This tool performs a single LLM call to explore multiple plausible directions
|
|
for a topic and briefly justify each one.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
_SYSTEM_PROMPT = """\
|
|
You are a breadth-first brainstorming engine.
|
|
|
|
Given a topic and optional supporting context, explore multiple promising
|
|
directions instead of converging too early on one answer.
|
|
|
|
Requirements:
|
|
- Generate 5-8 distinct possibilities from different angles when possible.
|
|
- Keep each possibility concrete and easy to scan.
|
|
- For each possibility, include a short rationale explaining why it is worth exploring.
|
|
- Prefer variety: methods, framing, applications, risks, experiments, or product directions.
|
|
- Do not pretend uncertain facts are verified.
|
|
- Keep the response concise, structured, and actionable.
|
|
|
|
Output in Markdown using this structure:
|
|
|
|
# Brainstorm
|
|
|
|
## 1. <short title>
|
|
- Direction: <1-2 sentence idea>
|
|
- Rationale: <brief why>
|
|
|
|
## 2. <short title>
|
|
- Direction: <1-2 sentence idea>
|
|
- Rationale: <brief why>
|
|
|
|
Continue for the remaining ideas.
|
|
"""
|
|
|
|
|
|
async def brainstorm(
|
|
topic: str,
|
|
context: str = "",
|
|
api_key: str | None = None,
|
|
base_url: str | None = None,
|
|
model: str | None = None,
|
|
max_tokens: int | None = None,
|
|
temperature: float | None = None,
|
|
) -> dict[str, Any]:
|
|
"""Generate breadth-first ideas for a topic via one LLM call."""
|
|
from deeptutor.services.config import get_agent_params
|
|
from deeptutor.services.llm import get_token_limit_kwargs
|
|
from deeptutor.services.llm import stream as llm_stream
|
|
from deeptutor.services.llm.config import get_llm_config
|
|
|
|
try:
|
|
llm_cfg = get_llm_config()
|
|
api_key = api_key or llm_cfg.api_key
|
|
base_url = base_url or llm_cfg.base_url
|
|
model = model or llm_cfg.model
|
|
except ValueError:
|
|
pass
|
|
|
|
if not model:
|
|
raise ValueError("No model configured for brainstorm tool")
|
|
|
|
agent_params = get_agent_params("brainstorm")
|
|
if max_tokens is None:
|
|
max_tokens = agent_params.get("max_tokens", 2048)
|
|
if temperature is None:
|
|
temperature = agent_params.get("temperature", 0.8)
|
|
|
|
parts: list[str] = [f"## Topic\n{topic.strip()}"]
|
|
if context and context.strip():
|
|
parts.append(f"## Context\n{context.strip()}")
|
|
user_prompt = "\n\n".join(parts)
|
|
|
|
kwargs: dict[str, Any] = {"temperature": temperature}
|
|
if max_tokens:
|
|
kwargs.update(get_token_limit_kwargs(model, max_tokens))
|
|
|
|
logger.debug("brainstorm tool: model=%s, topic=%s...", model, topic[:80])
|
|
|
|
_chunks: list[str] = []
|
|
async for _c in llm_stream(
|
|
prompt=user_prompt,
|
|
system_prompt=_SYSTEM_PROMPT,
|
|
model=model,
|
|
api_key=api_key,
|
|
base_url=base_url,
|
|
**kwargs,
|
|
):
|
|
_chunks.append(_c)
|
|
answer = "".join(_chunks)
|
|
|
|
return {
|
|
"topic": topic,
|
|
"answer": answer.strip(),
|
|
"model": model,
|
|
}
|