34 lines
892 B
C#
34 lines
892 B
C#
using System.Text.Json.Serialization;
|
|
|
|
namespace Toak.Api.Models;
|
|
|
|
public class LlamaRequestMessage
|
|
{
|
|
[JsonPropertyName("role")]
|
|
public string Role { get; set; } = string.Empty;
|
|
[JsonPropertyName("content")]
|
|
public string Content { get; set; } = string.Empty;
|
|
}
|
|
|
|
public class LlamaRequest
|
|
{
|
|
[JsonPropertyName("model")]
|
|
public string Model { get; set; } = "llama-3.1-8b-instant";
|
|
[JsonPropertyName("messages")]
|
|
public LlamaRequestMessage[] Messages { get; set; } = Array.Empty<LlamaRequestMessage>();
|
|
[JsonPropertyName("temperature")]
|
|
public double Temperature { get; set; } = 0.0;
|
|
}
|
|
|
|
public class LlamaResponse
|
|
{
|
|
[JsonPropertyName("choices")]
|
|
public LlamaChoice[] Choices { get; set; } = Array.Empty<LlamaChoice>();
|
|
}
|
|
|
|
public class LlamaChoice
|
|
{
|
|
[JsonPropertyName("message")]
|
|
public LlamaRequestMessage Message { get; set; } = new();
|
|
}
|