1230 lines
38 KiB
C#
1230 lines
38 KiB
C#
using System;
|
|
using OpenAI;
|
|
using OllamaSharp;
|
|
using System.Linq;
|
|
using xModels.Dtos;
|
|
using System.Net.Http;
|
|
using System.Threading;
|
|
using xAiApi.Constants;
|
|
using xAiModels.Models;
|
|
using xAiApi.Interfaces;
|
|
using xAiApi.Extensions;
|
|
using System.ClientModel;
|
|
using xCommons.Providers;
|
|
using xCommons.Extensions;
|
|
using xAiModels.Constants;
|
|
using xAiModels.Interfaces;
|
|
using xAiModels.Extensions;
|
|
using xAiService.Constants;
|
|
using xAiApi.Configurations;
|
|
using xExceptions.Constants;
|
|
using xAiModels.Models.Dtos;
|
|
using xAiService.Extensions;
|
|
using System.Threading.Tasks;
|
|
using xIdentityModels.Models;
|
|
using xFileService.Interfaces;
|
|
using Microsoft.Extensions.AI;
|
|
using Microsoft.AspNetCore.Http;
|
|
using System.Collections.Generic;
|
|
using Microsoft.Extensions.Logging;
|
|
using xAiApi.Interfaces.Extractors;
|
|
using System.ClientModel.Primitives;
|
|
using System.Runtime.CompilerServices;
|
|
|
|
namespace xAiApi.Providers
|
|
{
|
|
/// <summary>
|
|
/// Base Ai Service ...
|
|
/// </summary>
|
|
public abstract class XAIServiceBase : IXAiServiceBase
|
|
{
|
|
/// <summary>
|
|
/// Chat Options ...
|
|
/// </summary>
|
|
public ChatOptions Options { get; }
|
|
|
|
/// <summary>
|
|
/// Descriptor of Models which used in Service ...
|
|
/// </summary>
|
|
public XAiModelDescriptor Descriptor { get; }
|
|
|
|
private readonly IXFileProvider fileProvider;
|
|
private readonly IXAiDataProvider dataProvider;
|
|
private readonly ILogger<XAIServiceBase> logger;
|
|
private readonly XAiApiConfiguration configuration;
|
|
private readonly XValidationProvider validationProvider;
|
|
private readonly IXFileContentExtractor fileContentExtractor;
|
|
|
|
protected XAIServiceBase(
|
|
IXFileProvider fileProvider,
|
|
IXAiDataProvider dataProvider,
|
|
ILogger<XAIServiceBase> logger,
|
|
XAiApiConfiguration configuration,
|
|
XValidationProvider validationProvider,
|
|
IXFileContentExtractor fileContentExtractor,
|
|
string model = null
|
|
)
|
|
{
|
|
//
|
|
this.logger = logger;
|
|
this.fileProvider = fileProvider;
|
|
this.dataProvider = dataProvider;
|
|
this.configuration = configuration;
|
|
this.validationProvider = validationProvider;
|
|
this.fileContentExtractor = fileContentExtractor;
|
|
|
|
//
|
|
// Prepare Embedding Model Descriptor ...
|
|
Descriptor = configuration.GetModel(model);
|
|
if (!Descriptor.IsValid())
|
|
{
|
|
XException.InvalidConfiguration.Throw();
|
|
}
|
|
|
|
//
|
|
// Prepare Chat Options based On Tools ...
|
|
Options = new ChatOptions();
|
|
}
|
|
|
|
//
|
|
#region Actions ...
|
|
/// <summary>
|
|
/// Base Method for Communicate with LLM ...
|
|
/// </summary>
|
|
/// <param name="prompt"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public virtual async Task<string> AskTextAsync(
|
|
string prompt,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Check Prompt Exists ...
|
|
if (prompt.IsNullOrEmpty())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
using var client = GetClient();
|
|
var history = GetHistory(prompt);
|
|
var response = await client.GetResponseAsync(
|
|
messages: history,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// Validate Response ...
|
|
if (!response.IsValid())
|
|
{
|
|
//
|
|
// Dispose Client ...
|
|
client.Dispose();
|
|
XException.ActionFailed.Throw();
|
|
}
|
|
|
|
//
|
|
// Retrieve Response Text ...
|
|
var result = response.Text;
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Base Method for Communicate with LLM ...
|
|
/// </summary>
|
|
/// <param name="prompt"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public virtual async IAsyncEnumerable<string> AskTextAsEnumerable(
|
|
string prompt,
|
|
[EnumeratorCancellation]
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Check Prompt Exists ...
|
|
if (prompt.IsNullOrEmpty())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
using var client = GetClient();
|
|
var history = GetHistory(prompt);
|
|
var enumerable = client.GetStreamingResponseAsync(
|
|
options: null,
|
|
messages: history
|
|
);
|
|
|
|
//
|
|
await foreach (var res in enumerable)
|
|
{
|
|
//
|
|
// Cancellation Token ...
|
|
if (cancellationToken.IsCancellationRequested)
|
|
{
|
|
yield break;
|
|
}
|
|
|
|
//
|
|
yield return res.Text;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Get LLM Client instance for Communicating with LLM ...
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public virtual IChatClient GetClient()
|
|
{
|
|
//
|
|
// Try to Initialize LLm ...
|
|
var model = Descriptor.LLM;
|
|
var apiKey = Descriptor.ApiKey;
|
|
var url = new Uri(Descriptor.Url);
|
|
var httpClient = GetHttpClient(Descriptor.Url);
|
|
|
|
//
|
|
IChatClient result = null;
|
|
switch (Descriptor.Provider)
|
|
{
|
|
//
|
|
case XAiModelProviderType.Ollama:
|
|
//
|
|
// Instance Ollama Client using Custom HttpCLient ...
|
|
var ollamaClient = new OllamaApiClient(httpClient, model);
|
|
result =
|
|
new ChatClientBuilder(ollamaClient)
|
|
.UseFunctionInvocation()
|
|
.Build();
|
|
break;
|
|
|
|
//
|
|
case XAiModelProviderType.OpenAI:
|
|
//
|
|
// Instance Ollama Client using Custom HttpCLient ...
|
|
var openAiClient = new OpenAIClient(
|
|
new ApiKeyCredential(apiKey.IsNullOrEmpty() ? XAiApiConstants.XOpenAINoKey : apiKey),
|
|
new OpenAIClientOptions
|
|
{
|
|
Endpoint = url,
|
|
Transport = new HttpClientPipelineTransport(httpClient)
|
|
}
|
|
);
|
|
result =
|
|
new ChatClientBuilder(
|
|
openAiClient
|
|
.GetChatClient(model)
|
|
.AsIChatClient())
|
|
.UseFunctionInvocation()
|
|
.Build();
|
|
break;
|
|
|
|
//
|
|
case XAiModelProviderType.DeepSeek:
|
|
break;
|
|
|
|
//
|
|
case XAiModelProviderType.HuggingFace:
|
|
break;
|
|
|
|
//
|
|
default:
|
|
break;
|
|
}
|
|
|
|
//
|
|
if (result.IsNullOrDefault())
|
|
{
|
|
XException.InvalidData.Throw();
|
|
}
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Create Custom HttpClient for Communicating with LLM API ...
|
|
/// </summary>
|
|
/// <param name="url"></param>
|
|
/// <returns></returns>
|
|
public virtual HttpClient GetHttpClient(string url = null)
|
|
{
|
|
//
|
|
var result = new HttpClient
|
|
{
|
|
//
|
|
// Disable timeout completely (not recommended for production)
|
|
Timeout = Timeout.InfiniteTimeSpan,
|
|
|
|
//
|
|
BaseAddress = url.IsNullOrEmpty()
|
|
? null
|
|
: new Uri(url),
|
|
|
|
//
|
|
// OR set a very long timeout (e.g., 30 minutes for large file uploads)
|
|
// Timeout = TimeSpan.FromMinutes(30)
|
|
};
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Prepare a Message History List for LLM Communication based on Prompt ...
|
|
/// </summary>
|
|
/// <param name="prompt"></param>
|
|
/// <param name="forcePrompt"></param>
|
|
/// <returns></returns>
|
|
public virtual IList<ChatMessage> GetHistory(
|
|
string prompt = null,
|
|
bool forcePrompt = true
|
|
)
|
|
{
|
|
//
|
|
// Validate Args ...
|
|
if (prompt.IsNullOrEmpty())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
// User Prompt ...
|
|
var promptMessage = new ChatMessage(
|
|
ChatRole.User,
|
|
prompt
|
|
);
|
|
|
|
//
|
|
// Preparing Introduction Prompt ...
|
|
var introductionPrompt = new ChatMessage(
|
|
ChatRole.System,
|
|
configuration.GetPrompt()
|
|
);
|
|
|
|
//
|
|
List<ChatMessage> result =
|
|
[
|
|
introductionPrompt,
|
|
];
|
|
if (forcePrompt &&
|
|
prompt is not null &&
|
|
!string.IsNullOrWhiteSpace(prompt))
|
|
{
|
|
result.Add(promptMessage);
|
|
}
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Prepare a Message History List for LLM Communication based on Prompt ...
|
|
/// </summary>
|
|
/// <param name="message"></param>
|
|
/// <param name="forceMessage"></param>
|
|
/// <returns></returns>
|
|
public virtual IList<ChatMessage> GetHistory(
|
|
ChatMessage message = null,
|
|
bool forceMessage = true
|
|
)
|
|
{
|
|
//
|
|
// Preparing Introduction Prompt ...
|
|
var introductionPrompt = new ChatMessage(
|
|
ChatRole.System,
|
|
configuration.GetPrompt(XAiApiConstants.XAiApiIntroductionPromptName)
|
|
);
|
|
|
|
//
|
|
List<ChatMessage> result =
|
|
[
|
|
introductionPrompt,
|
|
];
|
|
if (forceMessage &&
|
|
!message.IsNullOrDefault())
|
|
{
|
|
result.Add(message);
|
|
}
|
|
|
|
//
|
|
return result;
|
|
}
|
|
#endregion
|
|
|
|
//
|
|
#region Loaders ...
|
|
/// <summary>
|
|
/// Load AI Projects ...
|
|
/// </summary>
|
|
/// <param name="ownerId"></param>
|
|
/// <param name="connectionId"></param>
|
|
/// <param name="includeDefaults"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async Task<IEnumerable<XAiProjectDto>> LoadProjects(
|
|
string ownerId,
|
|
string connectionId = null,
|
|
bool includeDefaults = false,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Validate Args ...
|
|
if (ownerId.IsNullOrEmpty())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
var result = await dataProvider.LoadProjects(
|
|
ownerId: ownerId,
|
|
fillConversations: false,
|
|
connectionId: connectionId,
|
|
fillConversationMessages: false,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// Handle Defaults Including ...
|
|
if (includeDefaults)
|
|
{
|
|
//
|
|
// Here we have to fill Conversations of Default Project ...
|
|
var defaultProjectId = result.GetDafualtProject(ownerId);
|
|
var defaultProject = result.FirstOrDefault(x => x.Id == defaultProjectId);
|
|
result = result.Where(x => x.Id != defaultProjectId);
|
|
|
|
//
|
|
// Reading Conversations of Specified Default Project ...
|
|
var defaultProjectConversations = await dataProvider.LoadConversations(
|
|
fillMessages: false,
|
|
connectionId: connectionId,
|
|
projectId: defaultProjectId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
var defaultConversationId = defaultProjectConversations.GetDefaultConversation(
|
|
ownerId: ownerId,
|
|
projectId: defaultProjectId
|
|
);
|
|
var defaultConversation = defaultProjectConversations.First(x => x.Id == defaultConversationId);
|
|
defaultProjectConversations = defaultProjectConversations.Where(x => x.Id != defaultConversationId);
|
|
|
|
//
|
|
// Reading Default Conversations Messages ...
|
|
var defaultConversationMessages = await dataProvider.LoadMessages(
|
|
cancellationToken: cancellationToken,
|
|
conversationId: defaultConversationId
|
|
);
|
|
|
|
//
|
|
defaultConversation.Messages = [.. defaultConversationMessages.OrderByDescending(m => m.CreatedOn)];
|
|
defaultProjectConversations = [defaultConversation, .. defaultProjectConversations];
|
|
defaultProject.Conversations = [.. defaultProjectConversations];
|
|
result = [defaultProject, .. result];
|
|
}
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Load Conversations of Specified AI Project ...
|
|
/// </summary>
|
|
/// <param name="projectId"></param>
|
|
/// <param name="connectionId"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async Task<IEnumerable<XAiConversationDto>> LoadConversations(
|
|
Guid projectId,
|
|
string connectionId = null,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Validate Args ...
|
|
if (projectId.IsDefaultGuid())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
var result = await dataProvider.LoadConversations(
|
|
fillMessages: false,
|
|
projectId: projectId,
|
|
connectionId: connectionId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Load Messages of Specified Conversation ...
|
|
/// </summary>
|
|
/// <param name="conversationId"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async Task<IEnumerable<XAiMessageDto>> LoadMessages(
|
|
Guid conversationId,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Validate Args ...
|
|
if (conversationId.IsDefaultGuid())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
var result = await dataProvider.LoadMessages(
|
|
conversationId: conversationId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
return result;
|
|
}
|
|
#endregion
|
|
|
|
//
|
|
#region Queries ...
|
|
/// <summary>
|
|
/// Query Projects ...
|
|
/// </summary>
|
|
/// <param name="ownerId"></param>
|
|
/// <param name="query"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async Task<XQueryResult<XAiProjectDto>> QueryProjects(
|
|
string ownerId,
|
|
XQuery query,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Validate ...
|
|
if (ownerId.IsNullOrEmpty())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
var result = await dataProvider.QueryProjects(
|
|
query: query,
|
|
ownerId: ownerId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Query Conversations ...
|
|
/// </summary>
|
|
/// <param name="ownerId"></param>
|
|
/// <param name="projectId"></param>
|
|
/// <param name="query"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async Task<XQueryResult<XAiConversationDto>> QueryConversations(
|
|
string ownerId,
|
|
Guid projectId,
|
|
XQuery query,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Validate ...
|
|
if (ownerId.IsNullOrEmpty() ||
|
|
projectId.IsDefaultGuid())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
// Retrieve Project ...
|
|
var project = await dataProvider.GetProject(
|
|
id: projectId,
|
|
fillConversations: false,
|
|
fillConversationMessages: false,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
if (project.IsNullOrDefault())
|
|
{
|
|
XException.NotFound.Throw();
|
|
}
|
|
|
|
//
|
|
// Validate Owner ...
|
|
if (ownerId != project.OwnerId)
|
|
{
|
|
XException.NotAllowed.Throw();
|
|
}
|
|
|
|
//
|
|
var result = await dataProvider.QueryConversations(
|
|
query: query,
|
|
ownerId: ownerId,
|
|
projectId: projectId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Query Messages ...
|
|
/// </summary>
|
|
/// <param name="ownerId"></param>
|
|
/// <param name="conversationId"></param>
|
|
/// <param name="query"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async Task<XQueryResult<XAiMessageDto>> QueryMessages(
|
|
string ownerId,
|
|
Guid conversationId,
|
|
XQuery query,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Validate ...
|
|
if (ownerId.IsNullOrEmpty() ||
|
|
conversationId.IsDefaultGuid())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
// Retrieve Conversation ...
|
|
var conversation = await dataProvider.GetConversation(
|
|
id: conversationId,
|
|
fillMessages: false,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
if (conversation.IsNullOrDefault())
|
|
{
|
|
XException.NotFound.Throw();
|
|
}
|
|
|
|
//
|
|
// Validate Owner ...
|
|
if (ownerId != conversation.OwnerId)
|
|
{
|
|
XException.NotAllowed.Throw();
|
|
}
|
|
|
|
//
|
|
var result = await dataProvider.QueryMessages(
|
|
query: query,
|
|
ownerId: ownerId,
|
|
conversationId: conversationId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
return result;
|
|
}
|
|
#endregion
|
|
|
|
//
|
|
#region Creators ...
|
|
/// <summary>
|
|
/// Create an AI Project ...
|
|
/// </summary>
|
|
/// <param name="ownerId"></param>
|
|
/// <param name="item"></param>
|
|
/// <param name="connectionId"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async Task<XAiProjectDto> CreateProject(
|
|
string ownerId,
|
|
XAiProjectDto item,
|
|
string connectionId = null,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
var result = await dataProvider.CreateProject(
|
|
item: item,
|
|
ownerId: ownerId,
|
|
connectionId: connectionId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Create an AI Conversation ...
|
|
/// </summary>
|
|
/// <param name="ownerId"></param>
|
|
/// <param name="projectId"></param>
|
|
/// <param name="item"></param>
|
|
/// <param name="connectionId"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async Task<XAiConversationDto> CreateConversation(
|
|
string ownerId,
|
|
Guid projectId,
|
|
XAiConversationDto item,
|
|
string connectionId = null,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
var result = await dataProvider.CreateConversation(
|
|
item: item,
|
|
ownerId: ownerId,
|
|
projectId: projectId,
|
|
connectionId: connectionId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
return result;
|
|
}
|
|
#endregion
|
|
|
|
//
|
|
#region Data Actions ...
|
|
/// <summary>
|
|
/// Ask a Question as Prompt and Retrieve Response ...
|
|
/// </summary>
|
|
/// <param name="prompt"></param>
|
|
/// <param name="ownerId"></param>
|
|
/// <param name="projectId"></param>
|
|
/// <param name="conversationId"></param>
|
|
/// <param name="connectionId"></param>
|
|
/// <param name="files"></param>
|
|
/// <param name="userInfo"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async Task<XAiMessageDto> AskAsync(
|
|
string prompt,
|
|
string ownerId,
|
|
Guid projectId,
|
|
Guid conversationId,
|
|
string connectionId = null,
|
|
IFormFileCollection files = null,
|
|
XUserClaimsInfoDto userInfo = null,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Validate Args ...
|
|
validationProvider
|
|
.GroupValidationBuilder()
|
|
.AddNotEmpty(prompt)
|
|
.AddNotEmpty(ownerId)
|
|
.ValidateGroup();
|
|
|
|
//
|
|
// Check Project ...
|
|
XAiProjectDto project = null;
|
|
if (projectId.IsDefaultGuid())
|
|
{
|
|
//
|
|
project = await dataProvider.GetDefaultProject(
|
|
ownerId: ownerId,
|
|
connectionId: connectionId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
projectId = project.Id;
|
|
}
|
|
else
|
|
{
|
|
//
|
|
project = await dataProvider.GetProject(
|
|
id: projectId,
|
|
fillConversations: false,
|
|
fillConversationMessages: false,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
}
|
|
if (project.IsNullOrDefault())
|
|
{
|
|
XException.NotFound.Throw();
|
|
}
|
|
|
|
//
|
|
// Check Conversation ...
|
|
XAiConversationDto conversation = null;
|
|
if (conversationId.IsDefaultGuid())
|
|
{
|
|
//
|
|
conversation = await dataProvider.GetDefaultConversation(
|
|
ownerId: ownerId,
|
|
projectId: projectId,
|
|
connectionId: connectionId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
conversationId = conversation.Id;
|
|
}
|
|
else
|
|
{
|
|
//
|
|
conversation = await dataProvider.GetConversation(
|
|
id: conversationId,
|
|
fillMessages: false,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
}
|
|
if (conversation.IsNullOrDefault())
|
|
{
|
|
XException.NotFound.Throw();
|
|
}
|
|
|
|
//
|
|
// Reading Messages ...
|
|
var messages = await dataProvider.LoadMessages(
|
|
conversationId: conversationId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// Upload Files ...
|
|
IEnumerable<xFileService.Models.Dtos.XFileDto> uploadedFiles = null;
|
|
if (files.HasChild())
|
|
{
|
|
//
|
|
// Validate User Info ...
|
|
if (userInfo.IsNullOrDefault())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
// Upload Files ...
|
|
uploadedFiles = await fileProvider.Upload(
|
|
files: files,
|
|
userInfo: userInfo,
|
|
connectionId: connectionId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
}
|
|
|
|
//
|
|
// Handle Memory ...
|
|
IList<ChatMessage> history = await PrepareMemory(
|
|
project: project,
|
|
messages: messages,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// Prepare and Add Prompt Message ...
|
|
var promptMessage = new XAiMessageDto
|
|
{
|
|
Content = prompt,
|
|
OwnerId = ownerId,
|
|
Role = XAiChatRole.User,
|
|
CreatedOn = DateTime.UtcNow,
|
|
ConversationId = conversationId,
|
|
ConversationTitle = conversation.Title
|
|
};
|
|
|
|
//
|
|
// Attached Uploaded File Id's as Meta Data ...
|
|
if (uploadedFiles.HasChild())
|
|
{
|
|
//
|
|
var fileIds = uploadedFiles.Select(f => f.Id).ToList();
|
|
promptMessage.MetaDatas = new Dictionary<string, object>
|
|
{
|
|
{ XAiServiceConstants.XMetaDataFileIdsIdentifier, fileIds }
|
|
}.ToJSON();
|
|
}
|
|
|
|
//
|
|
// Save Prompt Message ...
|
|
promptMessage = await dataProvider.AddMessage(
|
|
item: promptMessage,
|
|
connectionId: connectionId,
|
|
conversationId: conversationId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// Since Prompt Message Has Id now, we can Attach Uploaded Files to it ...
|
|
IList<AIContent> fileContents = null;
|
|
if (uploadedFiles.HasChild())
|
|
{
|
|
//
|
|
foreach (var uf in uploadedFiles)
|
|
{
|
|
//
|
|
await fileProvider.AddReference(
|
|
id: uf.Id,
|
|
userInfo: userInfo,
|
|
connectionId: connectionId,
|
|
forProvidedId: promptMessage.Id,
|
|
cancellationToken: cancellationToken,
|
|
providedFor: XAiServiceConstants.XAiMessageFilesIdentifier
|
|
);
|
|
}
|
|
|
|
//
|
|
// TODO: Extract File Contents ...
|
|
}
|
|
|
|
//
|
|
// Converts to ChatMessage ...
|
|
var promptChatMessage = promptMessage
|
|
.ToChatMessages(fileContents);
|
|
|
|
//
|
|
// Ask Questions From LLM ...
|
|
var answer = await AskLLMAsync(
|
|
history: history,
|
|
prompt: promptChatMessage,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// Prepare Answer AI Message ...
|
|
var result = new XAiMessageDto
|
|
{
|
|
Content = answer,
|
|
OwnerId = ownerId,
|
|
CreatedOn = DateTime.UtcNow,
|
|
Role = XAiChatRole.Assistant,
|
|
ConversationId = conversationId,
|
|
ConversationTitle = conversation.Title
|
|
};
|
|
result = await dataProvider.AddMessage(
|
|
item: result,
|
|
connectionId: connectionId,
|
|
conversationId: conversationId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Ask a Question as Prompt and Retrieve Response as Async Enumerable ...
|
|
/// </summary>
|
|
/// <param name="prompt"></param>
|
|
/// <param name="ownerId"></param>
|
|
/// <param name="projectId"></param>
|
|
/// <param name="conversationId"></param>
|
|
/// <param name="connectionId"></param>
|
|
/// <param name="files"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
public async IAsyncEnumerable<XAiResonseMessage> AskAsEnumerable(
|
|
string prompt,
|
|
string ownerId,
|
|
Guid projectId,
|
|
Guid conversationId,
|
|
string connectionId = null,
|
|
IFormFileCollection files = null,
|
|
[EnumeratorCancellation]
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
// Validate Args ...
|
|
validationProvider
|
|
.GroupValidationBuilder()
|
|
.AddNotEmpty(prompt)
|
|
.AddNotEmpty(ownerId)
|
|
.ValidateGroup();
|
|
|
|
//
|
|
// Check Project ...
|
|
XAiProjectDto project = null;
|
|
if (projectId.IsDefaultGuid())
|
|
{
|
|
//
|
|
project = await dataProvider.GetDefaultProject(
|
|
ownerId: ownerId,
|
|
connectionId: connectionId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
projectId = project.Id;
|
|
}
|
|
else
|
|
{
|
|
//
|
|
project = await dataProvider.GetProject(
|
|
id: projectId,
|
|
fillConversations: false,
|
|
fillConversationMessages: false,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
}
|
|
if (project.IsNullOrDefault())
|
|
{
|
|
XException.NotFound.Throw();
|
|
}
|
|
|
|
//
|
|
// Check Conversation ...
|
|
XAiConversationDto conversation = null;
|
|
if (conversationId.IsDefaultGuid())
|
|
{
|
|
//
|
|
conversation = await dataProvider.GetDefaultConversation(
|
|
ownerId: ownerId,
|
|
projectId: projectId,
|
|
connectionId: connectionId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
conversationId = conversation.Id;
|
|
}
|
|
else
|
|
{
|
|
//
|
|
conversation = await dataProvider.GetConversation(
|
|
id: conversationId,
|
|
fillMessages: false,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
}
|
|
if (conversation.IsNullOrDefault())
|
|
{
|
|
XException.NotFound.Throw();
|
|
}
|
|
|
|
//
|
|
// Reading Messages ...
|
|
var messages = await dataProvider.LoadMessages(
|
|
conversationId: conversationId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// TODO: Parse Files ...
|
|
// for Attached them into Prompt Message ...
|
|
|
|
//
|
|
// Handle Memory ...
|
|
IList<ChatMessage> history = await PrepareMemory(
|
|
project: project,
|
|
messages: messages,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// Prepare and Add Prompt Message ...
|
|
var promptMessage = new XAiMessageDto
|
|
{
|
|
Content = prompt,
|
|
OwnerId = ownerId,
|
|
Role = XAiChatRole.User,
|
|
CreatedOn = DateTime.UtcNow,
|
|
ConversationId = conversationId,
|
|
};
|
|
promptMessage = await dataProvider.AddMessage(
|
|
item: promptMessage,
|
|
connectionId: connectionId,
|
|
conversationId: conversationId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
promptMessage = await dataProvider.Enrich(
|
|
source: promptMessage,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// Attach Uploaded Files ...
|
|
var promptChatMessage = promptMessage
|
|
.ToChatMessages();
|
|
history.Add(promptChatMessage);
|
|
|
|
//
|
|
// Ask Questions From LLM ...
|
|
using var client = GetClient();
|
|
var enumerable = client.GetStreamingResponseAsync(
|
|
options: null,
|
|
messages: history
|
|
);
|
|
|
|
//
|
|
long sequence = 1;
|
|
var answer = string.Empty;
|
|
await foreach (var res in enumerable)
|
|
{
|
|
//
|
|
// Cancellation Token ...
|
|
if (cancellationToken.IsCancellationRequested)
|
|
{
|
|
yield break;
|
|
}
|
|
|
|
//
|
|
var resp = new XAiResonseMessage
|
|
{
|
|
OwnerId = ownerId,
|
|
Content = res.Text,
|
|
Sequence = sequence,
|
|
Role = XAiChatRole.System,
|
|
RespondTo = promptMessage.Id,
|
|
RespondedOn = DateTime.UtcNow,
|
|
};
|
|
|
|
//
|
|
sequence++;
|
|
answer += res.Text;
|
|
|
|
//
|
|
yield return resp;
|
|
}
|
|
|
|
//
|
|
// Prepare Answer AI Message ...
|
|
var result = new XAiMessageDto
|
|
{
|
|
Content = answer,
|
|
OwnerId = ownerId,
|
|
Role = XAiChatRole.System,
|
|
CreatedOn = DateTime.UtcNow,
|
|
ConversationId = conversationId
|
|
};
|
|
result = await dataProvider.AddMessage(
|
|
item: result,
|
|
connectionId: connectionId,
|
|
conversationId: conversationId,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
}
|
|
#endregion
|
|
|
|
/// <summary>
|
|
/// Base Dispose Action ...
|
|
/// </summary>
|
|
/// <exception cref="NotImplementedException"></exception>
|
|
public virtual void Dispose()
|
|
{ }
|
|
|
|
//
|
|
#region Private ...
|
|
/// <summary>
|
|
/// Implementing Memory Strategy here ...
|
|
/// </summary>
|
|
/// <param name="project"></param>
|
|
/// <param name="messages"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
private async Task<IList<ChatMessage>> PrepareMemory(
|
|
XAiProjectDto project,
|
|
IEnumerable<XAiMessageDto> messages,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
var result = new List<ChatMessage>();
|
|
|
|
//
|
|
// Retrieve Introduction Prompt ...
|
|
var introductionPrompt = configuration.GetPrompt(XAiApiConstants.XAiApiIntroductionPromptName);
|
|
if (introductionPrompt.IsNullOrEmpty())
|
|
{
|
|
XException.InvalidData.Throw();
|
|
}
|
|
|
|
//
|
|
var introductionMessage = new ChatMessage(
|
|
role: ChatRole.System,
|
|
content: introductionPrompt
|
|
);
|
|
|
|
//
|
|
result.Add(introductionMessage);
|
|
|
|
//
|
|
if (!project.IsNullOrDefault() &&
|
|
!project.Prompt.IsNullOrEmpty())
|
|
{
|
|
//
|
|
// TODO: Handle Project Files Here ...
|
|
var projectMessage = new ChatMessage(
|
|
role: ChatRole.System,
|
|
contents: [new TextContent(project.Prompt)]
|
|
);
|
|
result.Add(projectMessage);
|
|
}
|
|
|
|
//
|
|
if (messages.HasChild())
|
|
{
|
|
//
|
|
// TODO: Here we can Implement Memory Management based on
|
|
// Last nth of Messages, or Summarization of Messages, etc ...
|
|
result = [..result, ..messages
|
|
.OrderBy(m => m.CreatedOn)
|
|
.Select(m => m.ToChatMessages())];
|
|
}
|
|
|
|
//
|
|
return result;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Ask Quetion from LLM ...
|
|
/// </summary>
|
|
/// <param name="prompt"></param>
|
|
/// <param name="history"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
private async Task<string> AskLLMAsync(
|
|
ChatMessage prompt,
|
|
IList<ChatMessage> history,
|
|
CancellationToken cancellationToken = default
|
|
)
|
|
{
|
|
//
|
|
if (prompt.IsNullOrDefault())
|
|
{
|
|
XException.InvalidArgs.Throw();
|
|
}
|
|
|
|
//
|
|
history.Add(prompt);
|
|
using var client = GetClient();
|
|
var response = await client.GetResponseAsync(
|
|
messages: history,
|
|
cancellationToken: cancellationToken
|
|
);
|
|
|
|
//
|
|
// Validate Response ...
|
|
if (!response.IsValid())
|
|
{
|
|
//
|
|
// Dispose Client ...
|
|
client.Dispose();
|
|
XException.ActionFailed.Throw();
|
|
}
|
|
|
|
//
|
|
// Retrieve Response Text ...
|
|
var result = response.Text;
|
|
|
|
//
|
|
return result;
|
|
}
|
|
#endregion
|
|
}
|
|
} |