Выведите тип возвращаемого значения из параметра и карты

Чтобы это работало, вам нужно, чтобы OnToolCall<T> была универсальной функцией, поэтому тип возвращаемого значения OnToolCall<T> может зависеть от типа подсвойства toolName в аргументе. Этого не происходит, просто ссылаясь на тип типа typeof args['toolCall']['toolName']; если args относится только к типу { toolCall: { toolName: string} }, то этот тип — многословный способ сказать string. От таких дженериков не уйти.

Допустим, тип этого свойства toolName является параметром универсального типа K , ограниченным до keyof T. Тогда OnToolCall<T> следует определить следующим образом:

type OnToolCall<T extends Record<string, Tool>> =
  <K extends keyof T>(args: { toolCall: { toolName: K } }) =>
    ToolCallWithInferredReturnType<T>[K];

Теперь вы можете убедиться, что вызывающие абоненты видят желаемое поведение:

type Tools = {
  weather: {
    returnType: string;
  }
  cityAttractions: {
    returnType: string[];
  }
}

let onToolCall: OnToolCall<Tools> = ⋯;

const strArr = onToolCall({ toolCall: { toolName: "cityAttractions" } });
//    ^? const strArr: string[]
const str = onToolCall({ toolCall: { toolName: "weather" } });
//    ^? const str: string

Так что это выглядит хорошо.

Но учтите, что вы не можете реализовать onToolCall так, как делали раньше, по крайней мере, без утверждения типа :

onToolCall = ({ toolCall }) => { // error!
//~~~~~~~~ <-- Type 'string[] | string' is not assignable to 
//  type 'ToolCallWithInferredReturnType<Tools>[K]'
  if (toolCall.toolName === 'weather') {
    return 'Weather information was shown to the user.';
  }
  if (toolCall.toolName === 'cityAttractions') {
    return ["Attractions was shown to the user."]; 
  }
  throw Error("")
}

TypeScript не может использовать сужение потока управления (например, блоки if/else) для воздействия на параметры универсального типа. Вы можете проверить toolCall.toolName === 'weather', и это сузит toolCall.toolName, но это вообще не повлияет на параметр универсального типа K. Таким образом, TypeScript не может проверить, что string можно назначить ToolCallWithInferredReturnType<Tools>[K].

В microsoft/TypeScript#33014 уже давно существует запрос на открытую функцию, позволяющую этому работать, и есть некоторый прогресс в его реализации, но в настоящее время (начиная с TypeScript 5.5) это не является частью языка.

Возможно, в какой-нибудь будущей версии TypeScript приведенный выше код начнет работать как есть. Прямо сейчас, если вы хотите написать функцию, которая возвращает общий индексированный тип доступа, вам нужно будет реализовать ее путем индексации в объект. Так:

onToolCall = ({ toolCall }) => ({
  weather: 'Weather information was shown to the user.',
  cityAttractions: ["Attractions was shown to the user."]
}[toolCall.toolName]); // okay

По сути, это то же самое, что и предыдущая реализация, но здесь выполняется поиск вместо if/else. И если вы реализуете его неправильно, вы получите ожидаемую ошибку:

onToolCall = ({ toolCall }) => ({
  weather: 'Weather information was shown to the user.',
  cityAttractions: "Attractions was shown to the user."
}[toolCall.toolName]) // error!
//   Type 'string' is not assignable to type 'string[]'.    

хотя эта ошибка отображается во всем выражении. Если вам нужна ошибка, источник которой легче увидеть, вы можете записать ее в несколько операторов:

onToolCall = ({ toolCall }) => {
  const t: ToolCallWithInferredReturnType<Tools> = {
    weather: 'Weather information was shown to the user.',
    cityAttractions: "Attractions was shown to the user." // error!
    //~~~~~~~~~~~~~ <--
    // Type 'string' is not assignable to type 'string[]'.
  }
  return t[toolCall.toolName]
}

Детская площадка, ссылка на код