Skip to content

Commit

Permalink
WIP rework everything
Browse files Browse the repository at this point in the history
  • Loading branch information
PopDaph committed Sep 8, 2023
1 parent 3fdbe73 commit b20d025
Show file tree
Hide file tree
Showing 15 changed files with 1,085 additions and 1,241 deletions.
25 changes: 12 additions & 13 deletions front/lib/api/assistant/actions/retrieval.ts
Original file line number Diff line number Diff line change
Expand Up @@ -17,17 +17,17 @@ import { Err, Ok, Result } from "@app/lib/result";
import { new_id } from "@app/lib/utils";
import logger from "@app/logger/logger";
import {
AgentDataSourceConfigurationType,
isRetrievalConfiguration,
RetrievalActionType,
RetrievalConfigurationType,
RetrievalDocumentType,
TimeFrame,
} from "@app/types/assistant/actions/retrieval";
import { AgentActionSpecification } from "@app/types/assistant/agent";
import {
AgentActionSpecification,
AgentFullConfigurationType,
} from "@app/types/assistant/agent";
AgentConfigurationType,
AgentDataSourceConfigurationType,
isRetrievalConfiguration,
RetrievalConfigurationType,
} from "@app/types/assistant/configuration";
import {
AgentMessageType,
ConversationType,
Expand Down Expand Up @@ -167,7 +167,6 @@ export async function retrievalActionSpecification(
}

return {
id: configuration.id,
name: "search_data_sources",
description:
"Search the data sources specified by the user for information to answer their request." +
Expand Down Expand Up @@ -316,7 +315,7 @@ export type RetrievalSuccessEvent = {
// error is expected to be stored by the caller on the parent agent message.
export async function* runRetrieval(
auth: Authenticator,
configuration: AgentFullConfigurationType,
configuration: AgentConfigurationType,
conversation: ConversationType,
userMessage: UserMessageType,
agentMessage: AgentMessageType
Expand Down Expand Up @@ -349,7 +348,7 @@ export async function* runRetrieval(
return yield {
type: "retrieval_error",
created: Date.now(),
configurationId: configuration.agent.sId,
configurationId: configuration.sId,
messageId: agentMessage.sId,
error: {
code: "retrieval_parameters_generation_error",
Expand Down Expand Up @@ -433,7 +432,7 @@ export async function* runRetrieval(
return yield {
type: "retrieval_error",
created: Date.now(),
configurationId: configuration.agent.sId,
configurationId: configuration.sId,
messageId: agentMessage.sId,
error: {
code: "retrieval_search_error",
Expand All @@ -450,7 +449,7 @@ export async function* runRetrieval(
return yield {
type: "retrieval_error",
created: Date.now(),
configurationId: configuration.agent.sId,
configurationId: configuration.sId,
messageId: agentMessage.sId,
error: {
code: "retrieval_search_error",
Expand Down Expand Up @@ -529,15 +528,15 @@ export async function* runRetrieval(
yield {
type: "retrieval_documents",
created: Date.now(),
configurationId: configuration.agent.sId,
configurationId: configuration.sId,
messageId: agentMessage.sId,
documents,
};

yield {
type: "retrieval_success",
created: Date.now(),
configurationId: configuration.agent.sId,
configurationId: configuration.sId,
messageId: agentMessage.sId,
action: {
id: action.id,
Expand Down
6 changes: 3 additions & 3 deletions front/lib/api/assistant/agent.ts
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,11 @@ import {
import { Authenticator } from "@app/lib/auth";
import { Err, Ok, Result } from "@app/lib/result";
import { generateModelSId } from "@app/lib/utils";
import { isRetrievalConfiguration } from "@app/types/assistant/actions/retrieval";
import { AgentActionSpecification } from "@app/types/assistant/agent";
import {
AgentActionSpecification,
AgentConfigurationType,
} from "@app/types/assistant/agent";
isRetrievalConfiguration,
} from "@app/types/assistant/configuration";
import {
AgentActionType,
AgentMessageType,
Expand Down
283 changes: 0 additions & 283 deletions front/lib/api/assistant/agent/agent_create.ts

This file was deleted.

Loading

0 comments on commit b20d025

Please sign in to comment.