liteagent_sdk_dart 0.4.0

liteagent_sdk_dart: ^0.4.0 copied to clipboard

example/example.dart

import 'package:liteagent_sdk_dart/liteagent_sdk_dart.dart';

Future<void> main() async {
  String baseUrl = "<BASE_URL>";
  String apiKey = "<API_KEY>";
  String llmApiKey = "<LLM_API_KEY>";
  String llmBaseUrl = "<LLM_BASE_URL>";
  String llmModel = "<LLM_MODEL>";
  String userPrompt = "hi";
  LiteAgentSDK liteAgent = LiteAgentSDK(baseUrl: baseUrl, apiKey: apiKey);

  Capability capability = Capability(
    llmConfig: LLMConfig(
      baseUrl: llmBaseUrl,
      apiKey: llmApiKey,
      model: llmModel,
    ),
    systemPrompt: "You are a helpful assistant.",
  );

  Session session = await liteAgent.initSession(capability: capability);
  UserTask userTaskDto = UserTask(
    content: [Content(type: ContentType.text, message: userPrompt)],
    isChunk: true,
  );
  AgentMessageHandler agentMessageHandler = AgentMessageHandlerImpl();
  await liteAgent.chat(session, userTaskDto, agentMessageHandler);
}

class AgentMessageHandlerImpl extends AgentMessageHandler {
  AgentMessageHandlerImpl() : super();

  @override
  Future<void> onDone() async {
    print("[onDone]");
  }

  @override
  Future<void> onError(Exception e) async {
    print("[onError]$e");
  }

  @override
  Future<void> onMessage(String sessionId, AgentMessage agentMessageDto) async {
    print("sessionId: $sessionId, agentMessage: ${agentMessageDto.toJson()}");
  }

  @override
  Future<void> onChunk(
    String sessionId,
    AgentMessageChunk agentMessageChunkDto,
  ) async {
    print(
      "sessionId: $sessionId, agentMessageChunk: ${agentMessageChunkDto.toJson()}",
    );
  }
}