Coverage for agentos/llm.py: 0%

1from dataclasses import dataclass 

2from enum import Enum 

3class MessageRole(Enum): SYSTEM='system'; USER='user'; ASSISTANT='assistant' 

4@dataclass 

5class Message: role:MessageRole=MessageRole.USER; content:str='' 

6@dataclass 

7class ToolParameter: name:str='' 

8@dataclass 

9class ToolFunction: name:str='' 

10@dataclass 

11class Tool: type:str='function' 

12@dataclass 

13class ToolCall: id:str='' 

14@dataclass 

15class CompletionChoice: text:str='' 

16@dataclass 

17class TokenUsage: prompt:int=0; completion:int=0 

18@dataclass 

19class CompletionUsage: pass 

20@dataclass 

21class CompletionResult: choices:list=None 

22@dataclass 

23class StreamChunk: content:str='' 

24class LLMProvider: pass 

25class OpenAIProvider(LLMProvider): pass 

26class DeepSeekProvider(LLMProvider): pass 

27class AnthropicProvider(LLMProvider): pass 

28def create_provider(name:str) -> LLMProvider: return OpenAIProvider()