Start a trace for an LLM run.
(
self,
serialized: Dict[str, Any],
prompts: List[str],
*,
run_id: UUID,
tags: Optional[List[str]] = None,
parent_run_id: Optional[UUID] = None,
metadata: Optional[Dict[str, Any]] = None,
name: Optional[str] = None,
**kwargs: Any,
)
| 78 | return chat_model_run |
| 79 | |
| 80 | def on_llm_start( |
| 81 | self, |
| 82 | serialized: Dict[str, Any], |
| 83 | prompts: List[str], |
| 84 | *, |
| 85 | run_id: UUID, |
| 86 | tags: Optional[List[str]] = None, |
| 87 | parent_run_id: Optional[UUID] = None, |
| 88 | metadata: Optional[Dict[str, Any]] = None, |
| 89 | name: Optional[str] = None, |
| 90 | **kwargs: Any, |
| 91 | ) -> Run: |
| 92 | """Start a trace for an LLM run.""" |
| 93 | llm_run = self._create_llm_run( |
| 94 | serialized=serialized, |
| 95 | prompts=prompts, |
| 96 | run_id=run_id, |
| 97 | parent_run_id=parent_run_id, |
| 98 | tags=tags, |
| 99 | metadata=metadata, |
| 100 | name=name, |
| 101 | **kwargs, |
| 102 | ) |
| 103 | self._start_trace(llm_run) |
| 104 | self._on_llm_start(llm_run) |
| 105 | return llm_run |
| 106 | |
| 107 | def on_llm_new_token( |
| 108 | self, |
no test coverage detected