Coverage for intelligence_toolkit/AI/classes.py: 100%

1# Copyright (c) 2024 Microsoft Corporation. All rights reserved. 

2# Licensed under the MIT license. See LICENSE file in the project. 

3# 

4 

5 

6from typing import Any 

7 

8 

9class LLMCallback: 

10 """Class for LLM callbacks.""" 

11 

12 def __init__(self): 

13 self.response = [] 

14 

15 def on_llm_new_token(self, token: str): 

16 """Handle when a new token is generated.""" 

17 self.response.append(token) 

18 

19 

20class VectorData: 

21 hash: str 

22 text: str 

23 vector: list[float] 

24 additional_details: dict[str, Any]