@aj-archipelago/cortex 1.3.5 → 1.3.7

package/helper-apps/cortex-realtime-voice-server/src/cortex/reason.ts

@@ -0,0 +1,29 @@
+import {type ChatMessage, type CortexVariables, getCortexResponse} from "./utils";
+
+const WRITE_QUERY = `
+query Reason($text: String, $contextId: String, $chatHistory: [MultiMessage], $aiName: String) {
+  sys_entity_continue(text: $text, contextId: $contextId, chatHistory: $chatHistory, aiName: $aiName, generatorPathway: "sys_generator_reasoning", voiceResponse: true) {
+    result
+    tool
+    errors
+    warnings
+  }
+}
+`
+
+export async function reason(contextId: string,
+                             aiName: string,
+                             chatHistory: ChatMessage[],
+                             text: string) {
+
+  const variables: CortexVariables = {
+    chatHistory,
+    contextId,
+    aiName,
+    text
+  }
+
+  const res = await getCortexResponse(variables, WRITE_QUERY);
+
+  return res.sys_entity_continue;
+}

package/helper-apps/cortex-realtime-voice-server/src/cortex/search.ts

@@ -0,0 +1,30 @@
+import {type ChatMessage, type CortexVariables, type DataSource, getCortexResponse} from "./utils";
+
+const SEARCH_QUERY = `
+query Search($text: String, $contextId: String, $chatHistory: [MultiMessage], $aiName: String, $dataSources: [String]) {
+  sys_entity_continue(text: $text, contextId: $contextId, chatHistory: $chatHistory, aiName: $aiName, dataSources: $dataSources, generatorPathway: "sys_generator_results", voiceResponse: true) {
+    result
+    tool
+    errors
+    warnings
+  }
+}
+`
+
+export async function search(contextId: string,
+                             aiName: string,
+                             chatHistory: ChatMessage[],
+                             dataSources: DataSource[],
+                             text: string) {
+  const variables: CortexVariables = {
+    chatHistory,
+    contextId,
+    aiName,
+    dataSources,
+    text
+  }
+
+  const res = await getCortexResponse(variables, SEARCH_QUERY);
+
+  return res.sys_entity_continue;
+}

package/helper-apps/cortex-realtime-voice-server/src/cortex/style.ts

@@ -0,0 +1,31 @@
+import {type ChatMessage, type CortexVariables, getCortexResponse} from "./utils";
+
+const STYLE_QUERY = `
+query Style($text: String, $contextId: String, $chatHistory: [MultiMessage], $aiName: String, $aiStyle: String) {
+  sys_generator_voice_sample(text: $text, contextId: $contextId, chatHistory: $chatHistory, aiName: $aiName, aiStyle: $aiStyle) {
+    result
+    tool
+    errors
+    warnings
+  }
+}
+`
+
+export async function style(contextId: string,
+                             aiName: string,
+                             aiStyle: string,
+                             chatHistory: ChatMessage[],
+                             text: string) {
+
+  const variables: CortexVariables = {
+    chatHistory,
+    contextId,
+    aiName,
+    aiStyle,
+    text
+  }
+
+  const res = await getCortexResponse(variables, STYLE_QUERY);
+
+  return res.sys_generator_voice_sample;
+}

package/helper-apps/cortex-realtime-voice-server/src/cortex/utils.ts

@@ -0,0 +1,95 @@
+import { logger } from '../utils/logger';
+
+function getCortexApiKey() {
+  if (process.env.NODE_ENV === 'production') {
+    return process.env.CORTEX_API_KEY || ''
+  } else if (process.env.NODE_ENV === 'test') {
+    return process.env.CORTEX_DEV_API_KEY || ''
+  }
+  return '';
+}
+
+function getCortexUrl() {
+  if (process.env.NODE_ENV === 'production') {
+    return 'https://cortex.aljazeera.com/graphql'
+  } else if (process.env.NODE_ENV === 'test') {
+    return 'https://cortex.aljazeera.com/dev/graphql';
+  }
+  return 'http://localhost:4000/graphql';
+}
+
+function getHeaders() {
+  const headers: HeadersInit = new Headers();
+  headers.set('accept', 'application/json');
+  headers.set('Content-Type', 'application/json');
+  headers.set('ocp-apim-subscription-key', getCortexApiKey());
+  return headers;
+}
+
+export type ChatMessage = { role: string, content: string }
+export type DataSource = "mydata" | "aja" | "aje" | "wires" | "bing"
+
+export const MemorySection = {
+  memorySelf: "memorySelf",
+  memoryUser: "memoryUser",
+  memoryTopics: "memoryTopics",
+  memoryDirectives: "memoryDirectives",
+  memoryAll: "memoryAll"
+} as const;
+
+export type MemorySection = typeof MemorySection[keyof typeof MemorySection];
+
+export type CortexVariables = {
+  contextId?: string,
+  aiName?: string,
+  aiStyle?: string,
+  chatHistory?: ChatMessage[],
+  text?: string,
+  useMemory?: boolean,
+  language?: string,
+  dataSources?: DataSource[];
+  section?: MemorySection;
+  width?: number;
+  height?: number;
+  size?: string;
+  style?: string;
+  priority?: number;
+  recentHours?: number;
+  numResults?: number;
+}
+
+function truncateBody(body: any): string {
+  const str = JSON.stringify(body);
+  if (str.length <= 5000) return str;
+  
+  const halfLength = 2500;
+  return str.substring(0, halfLength) + '...' + str.substring(str.length - halfLength);
+}
+
+export async function getCortexResponse(
+  variables: CortexVariables,
+  query: string) {
+  const headers = getHeaders();
+  const body = {
+    query,
+    variables
+  }
+  logger.log(`Cortex URL: ${getCortexUrl()}`);
+  // logger.log(`Cortex Body: ${truncateBody(body)}`);
+  // logger.log(`Cortex Headers: ${JSON.stringify(headers)}`);
+  const res = await fetch(getCortexUrl(), {
+    method: 'POST',
+    headers,
+    body: JSON.stringify(body),
+  });
+
+  if (!res.ok) {
+    logger.error('Failed to fetch data:', res);
+    throw new Error('Failed to fetch data')
+  }
+
+  const responseObject = await res.json();
+  // Debug logging can be enabled/disabled via logger's environment control
+  logger.debug('cortex response', responseObject);
+  return responseObject.data;
+}

package/helper-apps/cortex-realtime-voice-server/src/cortex/vision.ts

@@ -0,0 +1,34 @@
+import {type ChatMessage, type CortexVariables, getCortexResponse} from "./utils";
+
+export type MultiMessage = {
+  role: string;
+  content: string | string[];
+}
+
+const VISION_QUERY = `
+query Vision($text: String, $contextId: String, $chatHistory: [MultiMessage], $aiName: String) {
+  sys_generator_video_vision(text: $text, contextId: $contextId, chatHistory: $chatHistory, aiName: $aiName) {
+    result
+    tool
+    errors
+    warnings
+  }
+}
+`
+
+export async function vision(contextId: string,
+                             aiName: string,
+                             chatHistory: (ChatMessage | MultiMessage)[],
+                             text: string) {
+
+  const variables: Omit<CortexVariables, 'chatHistory'> & { chatHistory: (ChatMessage | MultiMessage)[] } = {
+    chatHistory,
+    contextId,
+    aiName,
+    text
+  }
+
+  const res = await getCortexResponse(variables as CortexVariables, VISION_QUERY);
+
+  return res.sys_generator_video_vision;
+}