@kalenkevich/agent_007 0.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/agent/agent.d.ts +17 -0
- package/dist/agent/agent.js +6 -0
- package/dist/agent/agent.js.map +1 -0
- package/dist/agent/agent_event.d.ts +108 -0
- package/dist/agent/agent_event.js +56 -0
- package/dist/agent/agent_event.js.map +1 -0
- package/dist/agent/agent_event_utils.d.ts +5 -0
- package/dist/agent/agent_event_utils.js +66 -0
- package/dist/agent/agent_event_utils.js.map +1 -0
- package/dist/agent/cli_agent/cli_agent.d.ts +41 -0
- package/dist/agent/cli_agent/cli_agent.js +324 -0
- package/dist/agent/cli_agent/cli_agent.js.map +1 -0
- package/dist/agent/cli_agent/system_prompt.d.ts +1 -0
- package/dist/agent/cli_agent/system_prompt.js +58 -0
- package/dist/agent/cli_agent/system_prompt.js.map +1 -0
- package/dist/agent/cli_agent.d.ts +34 -0
- package/dist/agent/cli_agent.js +91 -0
- package/dist/agent/cli_agent.js.map +1 -0
- package/dist/agent/planner_agent/planner_agent.d.ts +28 -0
- package/dist/agent/planner_agent/planner_agent.js +102 -0
- package/dist/agent/planner_agent/planner_agent.js.map +1 -0
- package/dist/agent/request_processor/basic_request_processor.d.ts +17 -0
- package/dist/agent/request_processor/basic_request_processor.js +28 -0
- package/dist/agent/request_processor/basic_request_processor.js.map +1 -0
- package/dist/agent/request_processor/compaction_processor.d.ts +17 -0
- package/dist/agent/request_processor/compaction_processor.js +118 -0
- package/dist/agent/request_processor/compaction_processor.js.map +1 -0
- package/dist/agent/request_processor/request_processor.d.ts +11 -0
- package/dist/agent/request_processor/request_processor.js +2 -0
- package/dist/agent/request_processor/request_processor.js.map +1 -0
- package/dist/cli/init_project_command_handler.d.ts +3 -0
- package/dist/cli/init_project_command_handler.js +58 -0
- package/dist/cli/init_project_command_handler.js.map +1 -0
- package/dist/cli/loader.d.ts +6 -0
- package/dist/cli/loader.js +27 -0
- package/dist/cli/loader.js.map +1 -0
- package/dist/cli/prompt_utils.d.ts +11 -0
- package/dist/cli/prompt_utils.js +18 -0
- package/dist/cli/prompt_utils.js.map +1 -0
- package/dist/cli/run_command.d.ts +6 -0
- package/dist/cli/run_command.js +72 -0
- package/dist/cli/run_command.js.map +1 -0
- package/dist/cli/run_interactive_command.d.ts +6 -0
- package/dist/cli/run_interactive_command.js +282 -0
- package/dist/cli/run_interactive_command.js.map +1 -0
- package/dist/cli/run_noninteractive_command.d.ts +6 -0
- package/dist/cli/run_noninteractive_command.js +60 -0
- package/dist/cli/run_noninteractive_command.js.map +1 -0
- package/dist/cli_entrypoint.d.ts +2 -0
- package/dist/cli_entrypoint.js +110 -0
- package/dist/cli_entrypoint.js.map +1 -0
- package/dist/command/commnad_handler.d.ts +3 -0
- package/dist/command/commnad_handler.js +2 -0
- package/dist/command/commnad_handler.js.map +1 -0
- package/dist/command/init_project_command_handler.d.ts +4 -0
- package/dist/command/init_project_command_handler.js +58 -0
- package/dist/command/init_project_command_handler.js.map +1 -0
- package/dist/config/app_dir.d.ts +1 -0
- package/dist/config/app_dir.js +4 -0
- package/dist/config/app_dir.js.map +1 -0
- package/dist/config/config.d.ts +21 -0
- package/dist/config/config.js +2 -0
- package/dist/config/config.js.map +1 -0
- package/dist/config/config_loader.d.ts +2 -0
- package/dist/config/config_loader.js +47 -0
- package/dist/config/config_loader.js.map +1 -0
- package/dist/config/config_store.d.ts +10 -0
- package/dist/config/config_store.js +44 -0
- package/dist/config/config_store.js.map +1 -0
- package/dist/content.d.ts +51 -0
- package/dist/content.js +16 -0
- package/dist/content.js.map +1 -0
- package/dist/core/functions.d.ts +1 -0
- package/dist/core/functions.js +6 -0
- package/dist/core/functions.js.map +1 -0
- package/dist/core/loop.d.ts +20 -0
- package/dist/core/loop.js +121 -0
- package/dist/core/loop.js.map +1 -0
- package/dist/core/project_service.d.ts +11 -0
- package/dist/core/project_service.js +26 -0
- package/dist/core/project_service.js.map +1 -0
- package/dist/core/run.d.ts +6 -0
- package/dist/core/run.js +25 -0
- package/dist/core/run.js.map +1 -0
- package/dist/index.d.ts +1 -0
- package/dist/index.js +2 -0
- package/dist/index.js.map +1 -0
- package/dist/logger.d.ts +27 -0
- package/dist/logger.js +98 -0
- package/dist/logger.js.map +1 -0
- package/dist/model/adaptive_model.d.ts +16 -0
- package/dist/model/adaptive_model.js +57 -0
- package/dist/model/adaptive_model.js.map +1 -0
- package/dist/model/google/gemini_model.d.ts +15 -0
- package/dist/model/google/gemini_model.js +131 -0
- package/dist/model/google/gemini_model.js.map +1 -0
- package/dist/model/google/gemini_response_utils.d.ts +3 -0
- package/dist/model/google/gemini_response_utils.js +36 -0
- package/dist/model/google/gemini_response_utils.js.map +1 -0
- package/dist/model/google/gemini_streaming_utils.d.ts +28 -0
- package/dist/model/google/gemini_streaming_utils.js +235 -0
- package/dist/model/google/gemini_streaming_utils.js.map +1 -0
- package/dist/model/google/gen_ai_convert_utils.d.ts +6 -0
- package/dist/model/google/gen_ai_convert_utils.js +117 -0
- package/dist/model/google/gen_ai_convert_utils.js.map +1 -0
- package/dist/model/model.d.ts +11 -0
- package/dist/model/model.js +2 -0
- package/dist/model/model.js.map +1 -0
- package/dist/model/registry.d.ts +19 -0
- package/dist/model/registry.js +47 -0
- package/dist/model/registry.js.map +1 -0
- package/dist/model/request.d.ts +13 -0
- package/dist/model/request.js +2 -0
- package/dist/model/request.js.map +1 -0
- package/dist/model/request_builder_utils.d.ts +19 -0
- package/dist/model/request_builder_utils.js +43 -0
- package/dist/model/request_builder_utils.js.map +1 -0
- package/dist/model/response.d.ts +60 -0
- package/dist/model/response.js +2 -0
- package/dist/model/response.js.map +1 -0
- package/dist/model/util_llm.d.ts +10 -0
- package/dist/model/util_llm.js +149 -0
- package/dist/model/util_llm.js.map +1 -0
- package/dist/session/session.d.ts +14 -0
- package/dist/session/session.js +2 -0
- package/dist/session/session.js.map +1 -0
- package/dist/session/session_file_service.d.ts +23 -0
- package/dist/session/session_file_service.js +147 -0
- package/dist/session/session_file_service.js.map +1 -0
- package/dist/skills/skill.d.ts +23 -0
- package/dist/skills/skill.js +12 -0
- package/dist/skills/skill.js.map +1 -0
- package/dist/tools/build_in/find.d.ts +3 -0
- package/dist/tools/build_in/find.js +71 -0
- package/dist/tools/build_in/find.js.map +1 -0
- package/dist/tools/build_in/grep.d.ts +3 -0
- package/dist/tools/build_in/grep.js +97 -0
- package/dist/tools/build_in/grep.js.map +1 -0
- package/dist/tools/build_in/index.d.ts +1 -0
- package/dist/tools/build_in/index.js +13 -0
- package/dist/tools/build_in/index.js.map +1 -0
- package/dist/tools/build_in/list_dir.d.ts +3 -0
- package/dist/tools/build_in/list_dir.js +46 -0
- package/dist/tools/build_in/list_dir.js.map +1 -0
- package/dist/tools/build_in/view_file.d.ts +3 -0
- package/dist/tools/build_in/view_file.js +44 -0
- package/dist/tools/build_in/view_file.js.map +1 -0
- package/dist/tools/build_in/write_file.d.ts +3 -0
- package/dist/tools/build_in/write_file.js +50 -0
- package/dist/tools/build_in/write_file.js.map +1 -0
- package/dist/tools/functional_tool.d.ts +18 -0
- package/dist/tools/functional_tool.js +20 -0
- package/dist/tools/functional_tool.js.map +1 -0
- package/dist/tools/schema.d.ts +86 -0
- package/dist/tools/schema.js +37 -0
- package/dist/tools/schema.js.map +1 -0
- package/dist/tools/tool.d.ts +30 -0
- package/dist/tools/tool.js +9 -0
- package/dist/tools/tool.js.map +1 -0
- package/dist/tools/tool_call_policy.d.ts +4 -0
- package/dist/tools/tool_call_policy.js +4 -0
- package/dist/tools/tool_call_policy.js.map +1 -0
- package/dist/tools/tool_policy.d.ts +4 -0
- package/dist/tools/tool_policy.js +4 -0
- package/dist/tools/tool_policy.js.map +1 -0
- package/dist/ui/ui.d.ts +1 -0
- package/dist/ui/ui.js +2 -0
- package/dist/ui/ui.js.map +1 -0
- package/dist/user_input.d.ts +20 -0
- package/dist/user_input.js +21 -0
- package/dist/user_input.js.map +1 -0
- package/package.json +44 -0
- package/src/agent/agent.ts +18 -0
- package/src/agent/agent_event.ts +171 -0
- package/src/agent/agent_event_utils.ts +87 -0
- package/src/agent/cli_agent/cli_agent.ts +418 -0
- package/src/agent/cli_agent/system_prompt.ts +57 -0
- package/src/agent/planner_agent/planner_agent.ts +136 -0
- package/src/agent/request_processor/basic_request_processor.ts +46 -0
- package/src/agent/request_processor/compaction_processor.ts +164 -0
- package/src/agent/request_processor/request_processor.ts +13 -0
- package/src/cli/loader.ts +27 -0
- package/src/cli/prompt_utils.ts +19 -0
- package/src/cli/run_interactive_command.ts +337 -0
- package/src/cli/run_noninteractive_command.ts +74 -0
- package/src/cli_entrypoint.ts +128 -0
- package/src/command/commnad_handler.ts +3 -0
- package/src/command/init_project_command_handler.ts +66 -0
- package/src/config/app_dir.ts +4 -0
- package/src/config/config.ts +24 -0
- package/src/config/config_loader.ts +57 -0
- package/src/config/config_store.ts +50 -0
- package/src/content.ts +87 -0
- package/src/core/functions.ts +7 -0
- package/src/core/loop.ts +165 -0
- package/src/core/project_service.ts +38 -0
- package/src/core/run.ts +36 -0
- package/src/index.ts +1 -0
- package/src/logger.ts +128 -0
- package/src/model/adaptive_model.ts +77 -0
- package/src/model/google/gemini_model.ts +194 -0
- package/src/model/google/gemini_response_utils.ts +46 -0
- package/src/model/google/gemini_streaming_utils.ts +294 -0
- package/src/model/google/gen_ai_convert_utils.ts +149 -0
- package/src/model/model.ts +18 -0
- package/src/model/registry.ts +61 -0
- package/src/model/request.ts +15 -0
- package/src/model/request_builder_utils.ts +79 -0
- package/src/model/response.ts +66 -0
- package/src/model/util_llm.ts +167 -0
- package/src/session/session.ts +16 -0
- package/src/session/session_file_service.ts +207 -0
- package/src/skills/skill.ts +38 -0
- package/src/tools/build_in/find.ts +80 -0
- package/src/tools/build_in/grep.ts +101 -0
- package/src/tools/build_in/index.ts +13 -0
- package/src/tools/build_in/list_dir.ts +50 -0
- package/src/tools/build_in/view_file.ts +47 -0
- package/src/tools/build_in/write_file.ts +53 -0
- package/src/tools/functional_tool.ts +59 -0
- package/src/tools/schema.ts +87 -0
- package/src/tools/tool.ts +68 -0
- package/src/tools/tool_call_policy.ts +7 -0
- package/src/ui/ui.ts +0 -0
- package/src/user_input.ts +51 -0
- package/tests/integration/util_llm_test.ts +42 -0
- package/tests/unit/adaptive_model_test.ts +122 -0
- package/tests/unit/agent/request_processor/compaction_processor_test.ts +121 -0
- package/tests/unit/cli/prompt_utils_test.ts +47 -0
- package/tests/unit/cli_agent_test.ts +476 -0
- package/tests/unit/content_test.ts +56 -0
- package/tests/unit/logger_test.ts +109 -0
- package/tests/unit/loop_test.ts +141 -0
- package/tests/unit/model/gemini_model_test.ts +111 -0
- package/tests/unit/planner_agent_test.ts +52 -0
- package/tests/unit/project_service_test.ts +102 -0
- package/tests/unit/session_file_service_test.ts +160 -0
- package/tests/unit/tools/find_test.ts +40 -0
- package/tests/unit/tools/grep_test.ts +43 -0
- package/tests/unit/tools/list_dir_test.ts +32 -0
- package/tests/unit/tools/view_file_test.ts +32 -0
- package/tests/unit/tools/write_file_test.ts +44 -0
- package/tests/unit/user_input_test.ts +55 -0
- package/tests/unit/util_llm_test.ts +61 -0
- package/todo.md +29 -0
- package/tsconfig.json +33 -0
- package/vitest.config.ts +33 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini_streaming_utils.js","sourceRoot":"","sources":["../../../src/model/google/gemini_streaming_utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AACtE,OAAO,EAAE,4BAA4B,EAAE,MAAM,yBAAyB,CAAC;AAEvE,OAAO,EAIL,iBAAiB,EACjB,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAE/D;;;;;GAKG;AACH,MAAM,OAAO,2BAA2B;IAAxC;QAME,gEAAgE;QACxD,kBAAa,GAAkB,EAAE,CAAC;QAClC,sBAAiB,GAAG,EAAE,CAAC;QAMvB,kBAAa,GAA4B,EAAE,CAAC;IAoQtD,CAAC;IAhQS,yBAAyB;QAC/B,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;gBACtB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,IAAI,CAAC,iBAAiB;aAChC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,iBAAiB;aAC7B,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAC;IACxC,CAAC;IAEO,sBAAsB,CAC5B,UAAsB,EACtB,QAAgB;QAEhB,IAAI,KAAK,GAAY,IAAI,CAAC;QAC1B,IAAI,QAAQ,GAAG,KAAK,CAAC;QAErB,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;QAC3C,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;QAC3C,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACvC,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QAEvC,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC9B,MAAM,WAAW,GAAG,WAAW,CAAC;YAChC,QAAQ,GAAG,IAAI,CAAC;YAEhB,MAAM,iBAAiB,GAAG,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC;gBACjD,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;gBACnB,CAAC,CAAC,QAAQ,CAAC;YACb,MAAM,SAAS,GAAG,iBAAiB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAE/C,IAAI,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;YACvC,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;gBAC7B,IACE,aAAa;oBACb,OAAO,aAAa,KAAK,QAAQ;oBACjC,IAAI,IAAI,aAAa,EACrB,CAAC;oBACD,aAAa,GAAG,aAAa,CAAC,IAAI,CAA4B,CAAC;gBACjE,CAAC;qBAAM,CAAC;oBACN,MAAM;gBACR,CAAC;YACH,CAAC;YAED,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;gBACtC,KAAK,GAAG,aAAa,GAAG,WAAW,CAAC;YACtC,CAAC;iBAAM,CAAC;gBACN,KAAK,GAAG,WAAW,CAAC;YACtB,CAAC;QACH,CAAC;aAAM,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YACrC,KAAK,GAAG,WAAW,CAAC;YACpB,QAAQ,GAAG,IAAI,CAAC;QAClB,CAAC;aAAM,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACnC,KAAK,GAAG,SAAS,CAAC;YAClB,QAAQ,GAAG,IAAI,CAAC;QAClB,CAAC;aAAM,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACnC,KAAK,GAAG,IAAI,CAAC;YACb,QAAQ,GAAG,IAAI,CAAC;QAClB,CAAC;QAED,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC3B,CAAC;IAEO,kBAAkB,CAAC,QAAgB,EAAE,KAAc;QACzD,MAAM,IAAI,GAAG,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QACtE,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAElC,IAAI,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;YAC1B,IAAI,CAAC,CAAC,IAAI,IAAI,OAAO,CAAC,EAAE,CAAC;gBACvB,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;YACrB,CAAC;YACD,OAAO,GAAG,OAAO,CAAC,IAAI,CAA4B,CAAC;QACrD,CAAC;QAED,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;IACnD,CAAC;IAEO,2BAA2B;QACjC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,MAAM,MAAM,GAA4B;gBACtC,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,IAAI,CAAC,aAAa;gBACxB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACpD,EAAE,EAAE,IAAI,CAAC,WAAW,IAAI,4BAA4B,EAAE;aACvD,CAAC;YAEF,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;gBACjC,MAAM,CAAC,gBAAgB,GAAG,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,CAAC;YACpE,CAAC;YAED,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEhC,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;YAC/B,IAAI,CAAC,aAAa,GAAG,EAAE,CAAC;YACxB,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;YAC7B,IAAI,CAAC,uBAAuB,GAAG,SAAS,CAAC;QAC3C,CAAC;IACH,CAAC;IAEO,4BAA4B,CAAC,EAA2B;QAC9D,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YACZ,IAAI,CAAC,aAAa,GAAG,EAAE,CAAC,IAAI,CAAC;QAC/B,CAAC;QACD,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YACV,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC;QAC3B,CAAC;QAED,KAAK,MAAM,UAAU,IAAI,EAAE,CAAC,WAAW,IAAI,EAAE,EAAE,CAAC;YAC9C,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;YACrC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,SAAS;YACX,CAAC;YAED,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC,sBAAsB,CACnD,UAAU,EACV,QAAQ,CACT,CAAC;YAEF,IAAI,QAAQ,EAAE,CAAC;gBACb,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QAED,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC;YACrB,IAAI,CAAC,yBAAyB,EAAE,CAAC;YACjC,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,uBAAuB,CAAC,IAA6B;QAC3D,MAAM,EAAE,GAAG,IAAI,CAAC;QAEhB,IAAI,EAAE,CAAC,WAAW,IAAI,EAAE,CAAC,YAAY,EAAE,CAAC;YACtC,IAAI,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBAChC,EAAE,CAAC,EAAE,GAAG,4BAA4B,EAAE,CAAC;YACzC,CAAC;YAED,IAAI,IAAI,CAAC,gBAAgB,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC;gBAC3D,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,gBAAgB,CAAC;YACvD,CAAC;YACD,IAAI,CAAC,4BAA4B,CAAC,EAAE,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;gBACZ,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;oBACX,EAAE,CAAC,EAAE,GAAG,4BAA4B,EAAE,CAAC;gBACzC,CAAC;gBACD,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBACjC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,eAAe,CACpB,QAAiC;QAEjC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,MAAM,WAAW,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,CAAC,aAAa,GAAG,WAAW,CAAC,aAAa,CAAC;QAC/C,uCAAuC;QACvC,4DAA4D;QAC5D,IAAI;QACJ,sCAAsC;QACtC,0DAA0D;QAC1D,IAAI;QAEJ,IAAI,WAAW,CAAC,YAAY,EAAE,CAAC;YAC7B,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,YAAY,CAAC;QAC/C,CAAC;QAED,IAAI,WAAW,CAAC,OAAO,IAAI,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACrD,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC7C,IAAI,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC5B,MAAM,SAAS,GAAG,KAAK,CAAC;oBACxB,IACE,IAAI,CAAC,iBAAiB;wBACtB,SAAS,KAAK,IAAI,CAAC,oBAAoB,EACvC,CAAC;wBACD,IAAI,CAAC,yBAAyB,EAAE,CAAC;oBACnC,CAAC;oBAED,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;wBAC5B,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAC;oBACxC,CAAC;oBACD,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,IAAI,CAAC;gBACtC,CAAC;qBAAM,IAAI,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC;oBACtC,MAAM,SAAS,GAAG,IAAI,CAAC;oBACvB,IACE,IAAI,CAAC,iBAAiB;wBACtB,SAAS,KAAK,IAAI,CAAC,oBAAoB,EACvC,CAAC;wBACD,IAAI,CAAC,yBAAyB,EAAE,CAAC;oBACnC,CAAC;oBAED,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;wBAC5B,IAAI,CAAC,oBAAoB,GAAG,SAAS,CAAC;oBACxC,CAAC;oBACD,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,OAAO,CAAC;gBACzC,CAAC;qBAAM,IAAI,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC3C,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;gBACrC,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,yBAAyB,EAAE,CAAC;oBACjC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;QACH,CAAC;QAED,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;QAC3B,MAAM,WAAW,CAAC;IACpB,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,UAAU,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxE,OAAO;QACT,CAAC;QAED,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACjC,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAEnC,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC;QACtC,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,IAAI,SAAS,CAAC,YAAY,CAAC;QAEjE,OAAO;YACL,OAAO,EAAE;gBACP,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,UAAU;aAClB;YACD,6CAA6C;YAC7C,2CAA2C;YAC3C,SAAS,EAAE,YAAY,KAAK,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY;YACxE,YAAY,EACV,YAAY,KAAK,YAAY,CAAC,IAAI;gBAChC,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,SAAS,CAAC,aAAa;YAC7B,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,YAAY,EAAE,YAAY;YAC1B,OAAO,EAAE,KAAK;SACf,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import { type Content, type ContentPart } from "../../content.js";
|
|
2
|
+
import type { Part as GenAIContentPart, Content as GenAIContent } from "@google/genai";
|
|
3
|
+
export declare function contentToGenAIContent(content: Content): GenAIContent;
|
|
4
|
+
export declare function contentPartToGenAIContentPart(part: ContentPart): GenAIContentPart;
|
|
5
|
+
export declare function genAIContentToContent(genAIContent: GenAIContent): Content;
|
|
6
|
+
export declare function genAIContentPartToContentPart(part: GenAIContentPart): ContentPart;
|
|
@@ -0,0 +1,117 @@
|
|
|
1
|
+
import { isTextContentPart, isThoughtContentPart, isMediaContentPart, isFunctionCallContentPart, isFunctionResponseContentPart, } from "../../content.js";
|
|
2
|
+
export function contentToGenAIContent(content) {
|
|
3
|
+
return {
|
|
4
|
+
parts: content.parts.map((part) => contentPartToGenAIContentPart(part)),
|
|
5
|
+
role: content.role === "user" ? "user" : "model",
|
|
6
|
+
};
|
|
7
|
+
}
|
|
8
|
+
export function contentPartToGenAIContentPart(part) {
|
|
9
|
+
if (isTextContentPart(part)) {
|
|
10
|
+
return {
|
|
11
|
+
text: part.text,
|
|
12
|
+
};
|
|
13
|
+
}
|
|
14
|
+
if (isThoughtContentPart(part)) {
|
|
15
|
+
return {
|
|
16
|
+
text: part.thought,
|
|
17
|
+
thought: true,
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
if (isMediaContentPart(part) && part.data) {
|
|
21
|
+
return {
|
|
22
|
+
inlineData: {
|
|
23
|
+
data: part.data,
|
|
24
|
+
mimeType: part.mimeType,
|
|
25
|
+
},
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
if (isMediaContentPart(part) && part.uri) {
|
|
29
|
+
return {
|
|
30
|
+
fileData: {
|
|
31
|
+
fileUri: part.uri,
|
|
32
|
+
mimeType: part.mimeType,
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
if (isFunctionCallContentPart(part)) {
|
|
37
|
+
return {
|
|
38
|
+
functionCall: {
|
|
39
|
+
id: part.id,
|
|
40
|
+
name: part.name,
|
|
41
|
+
args: part.args,
|
|
42
|
+
partialArgs: part.partialArgs,
|
|
43
|
+
willContinue: part.willContinue,
|
|
44
|
+
},
|
|
45
|
+
thoughtSignature: part.thoughtSignature,
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
if (isFunctionResponseContentPart(part)) {
|
|
49
|
+
return {
|
|
50
|
+
functionResponse: {
|
|
51
|
+
id: part.id,
|
|
52
|
+
name: part.name,
|
|
53
|
+
response: part.response,
|
|
54
|
+
willContinue: part.willContinue,
|
|
55
|
+
},
|
|
56
|
+
thoughtSignature: part.thoughtSignature,
|
|
57
|
+
};
|
|
58
|
+
}
|
|
59
|
+
throw new Error(`Unsupported content type: ${part.type}`);
|
|
60
|
+
}
|
|
61
|
+
export function genAIContentToContent(genAIContent) {
|
|
62
|
+
return {
|
|
63
|
+
role: genAIContent.role === "user" ? "user" : "agent",
|
|
64
|
+
parts: (genAIContent.parts || []).map((p) => genAIContentPartToContentPart(p)),
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
export function genAIContentPartToContentPart(part) {
|
|
68
|
+
if (part.text !== undefined) {
|
|
69
|
+
if (part.thought) {
|
|
70
|
+
return {
|
|
71
|
+
type: "thought",
|
|
72
|
+
thought: part.text,
|
|
73
|
+
};
|
|
74
|
+
}
|
|
75
|
+
return {
|
|
76
|
+
type: "text",
|
|
77
|
+
text: part.text,
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
if (part.inlineData) {
|
|
81
|
+
return {
|
|
82
|
+
type: "media",
|
|
83
|
+
data: part.inlineData.data,
|
|
84
|
+
mimeType: part.inlineData.mimeType,
|
|
85
|
+
};
|
|
86
|
+
}
|
|
87
|
+
if (part.fileData) {
|
|
88
|
+
return {
|
|
89
|
+
type: "media",
|
|
90
|
+
uri: part.fileData.fileUri,
|
|
91
|
+
mimeType: part.fileData.mimeType,
|
|
92
|
+
};
|
|
93
|
+
}
|
|
94
|
+
if (part.functionCall) {
|
|
95
|
+
return {
|
|
96
|
+
type: "function_call",
|
|
97
|
+
id: part.functionCall.id,
|
|
98
|
+
name: part.functionCall.name,
|
|
99
|
+
args: part.functionCall.args,
|
|
100
|
+
partialArgs: part.functionCall.partialArgs,
|
|
101
|
+
willContinue: part.functionCall.willContinue,
|
|
102
|
+
thoughtSignature: part.thoughtSignature,
|
|
103
|
+
};
|
|
104
|
+
}
|
|
105
|
+
if (part.functionResponse) {
|
|
106
|
+
return {
|
|
107
|
+
type: "function_response",
|
|
108
|
+
id: part.functionResponse.id,
|
|
109
|
+
name: part.functionResponse.name,
|
|
110
|
+
response: part.functionResponse.response,
|
|
111
|
+
willContinue: part.functionResponse.willContinue,
|
|
112
|
+
thoughtSignature: part.thoughtSignature,
|
|
113
|
+
};
|
|
114
|
+
}
|
|
115
|
+
throw new Error(`Unsupported GenAI content part`);
|
|
116
|
+
}
|
|
117
|
+
//# sourceMappingURL=gen_ai_convert_utils.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gen_ai_convert_utils.js","sourceRoot":"","sources":["../../../src/model/google/gen_ai_convert_utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,iBAAiB,EACjB,oBAAoB,EACpB,kBAAkB,EAClB,yBAAyB,EACzB,6BAA6B,GAC9B,MAAM,kBAAkB,CAAC;AAM1B,MAAM,UAAU,qBAAqB,CAAC,OAAgB;IACpD,OAAO;QACL,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,6BAA6B,CAAC,IAAI,CAAC,CAAC;QACvE,IAAI,EAAE,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO;KACjD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC3C,IAAiB;IAEjB,IAAI,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5B,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;SAChB,CAAC;IACJ,CAAC;IAED,IAAI,oBAAoB,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/B,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,OAAO;YAClB,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,IAAI,kBAAkB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QAC1C,OAAO;YACL,UAAU,EAAE;gBACV,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB;SACF,CAAC;IACJ,CAAC;IAED,IAAI,kBAAkB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QACzC,OAAO;YACL,QAAQ,EAAE;gBACR,OAAO,EAAE,IAAI,CAAC,GAAG;gBACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB;SACF,CAAC;IACJ,CAAC;IAED,IAAI,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;QACpC,OAAO;YACL,YAAY,EAAE;gBACZ,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,YAAY,EAAE,IAAI,CAAC,YAAY;aAChC;YACD,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SACxC,CAAC;IACJ,CAAC;IAED,IAAI,6BAA6B,CAAC,IAAI,CAAC,EAAE,CAAC;QACxC,OAAO;YACL,gBAAgB,EAAE;gBAChB,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,YAAY,EAAE,IAAI,CAAC,YAAY;aAChC;YACD,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SACxC,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,6BAA6B,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;AAC5D,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,YAA0B;IAC9D,OAAO;QACL,IAAI,EAAE,YAAY,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO;QACrD,KAAK,EAAE,CAAC,YAAY,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC1C,6BAA6B,CAAC,CAAC,CAAC,CACjC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC3C,IAAsB;IAEtB,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QAC5B,IAAK,IAAY,CAAC,OAAO,EAAE,CAAC;YAC1B,OAAO;gBACL,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,IAAI,CAAC,IAAI;aACnB,CAAC;QACJ,CAAC;QACD,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,IAAI,CAAC,IAAI;SAChB,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,OAAO;YACL,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;YAC1B,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ;SACnC,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;QAClB,OAAO;YACL,IAAI,EAAE,OAAO;YACb,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO;YAC1B,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;SACjC,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;QACtB,OAAO;YACL,IAAI,EAAE,eAAe;YACrB,EAAE,EAAE,IAAI,CAAC,YAAY,CAAC,EAAE;YACxB,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;YAC5B,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;YAC5B,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,WAAW;YAC1C,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,YAAY;YAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SACxC,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1B,OAAO;YACL,IAAI,EAAE,mBAAmB;YACzB,EAAE,EAAE,IAAI,CAAC,gBAAgB,CAAC,EAAE;YAC5B,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI;YAChC,QAAQ,EAAE,IAAI,CAAC,gBAAgB,CAAC,QAAQ;YACxC,YAAY,EAAE,IAAI,CAAC,gBAAgB,CAAC,YAAY;YAChD,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SACxC,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;AACpD,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { LlmRequest } from "./request.js";
|
|
2
|
+
import type { LlmResponse } from "./response.js";
|
|
3
|
+
export interface LlmModelConfig {
|
|
4
|
+
stream?: boolean;
|
|
5
|
+
abortSignal?: AbortSignal;
|
|
6
|
+
}
|
|
7
|
+
export interface LlmModel {
|
|
8
|
+
modelName: string;
|
|
9
|
+
run(request: LlmRequest, config?: LlmModelConfig): AsyncGenerator<LlmResponse, void, unknown>;
|
|
10
|
+
countTokens(request: LlmRequest): Promise<number>;
|
|
11
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model.js","sourceRoot":"","sources":["../../src/model/model.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { LlmModel } from "./model.js";
|
|
2
|
+
type LlmModelClass = {
|
|
3
|
+
new (params: {
|
|
4
|
+
modelName: string;
|
|
5
|
+
apiKey: string;
|
|
6
|
+
}): LlmModel;
|
|
7
|
+
};
|
|
8
|
+
export declare enum ModelType {
|
|
9
|
+
LITE = "lite",
|
|
10
|
+
FAST = "fast",
|
|
11
|
+
PRO = "pro"
|
|
12
|
+
}
|
|
13
|
+
export interface ModelInfo {
|
|
14
|
+
name: string;
|
|
15
|
+
type: ModelType;
|
|
16
|
+
klass: LlmModelClass;
|
|
17
|
+
}
|
|
18
|
+
export declare function resolveLlmModel(modelName: string): LlmModelClass;
|
|
19
|
+
export {};
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
import { Gemini } from "./google/gemini_model.js";
|
|
2
|
+
export var ModelType;
|
|
3
|
+
(function (ModelType) {
|
|
4
|
+
ModelType["LITE"] = "lite";
|
|
5
|
+
ModelType["FAST"] = "fast";
|
|
6
|
+
ModelType["PRO"] = "pro";
|
|
7
|
+
})(ModelType || (ModelType = {}));
|
|
8
|
+
const MODEL_REGISTRY = {
|
|
9
|
+
"gemini-3.1-pro-preview": {
|
|
10
|
+
name: "gemini-3.1-pro-preview",
|
|
11
|
+
type: ModelType.PRO,
|
|
12
|
+
klass: Gemini,
|
|
13
|
+
},
|
|
14
|
+
"gemini-3-flash-preview": {
|
|
15
|
+
name: "gemini-3-flash-preview",
|
|
16
|
+
type: ModelType.FAST,
|
|
17
|
+
klass: Gemini,
|
|
18
|
+
},
|
|
19
|
+
"gemini-3.1-flash-lite-preview": {
|
|
20
|
+
name: "gemini-3.1-flash-lite-preview",
|
|
21
|
+
type: ModelType.LITE,
|
|
22
|
+
klass: Gemini,
|
|
23
|
+
},
|
|
24
|
+
"gemini-2.5-pro": {
|
|
25
|
+
name: "gemini-2.5-pro",
|
|
26
|
+
type: ModelType.PRO,
|
|
27
|
+
klass: Gemini,
|
|
28
|
+
},
|
|
29
|
+
"gemini-2.5-flash": {
|
|
30
|
+
name: "gemini-2.5-flash",
|
|
31
|
+
type: ModelType.FAST,
|
|
32
|
+
klass: Gemini,
|
|
33
|
+
},
|
|
34
|
+
"gemini-2.5-flash-lite": {
|
|
35
|
+
name: "gemini-2.5-flash-lite",
|
|
36
|
+
type: ModelType.LITE,
|
|
37
|
+
klass: Gemini,
|
|
38
|
+
},
|
|
39
|
+
};
|
|
40
|
+
export function resolveLlmModel(modelName) {
|
|
41
|
+
const modelInfo = MODEL_REGISTRY[modelName];
|
|
42
|
+
if (!modelInfo) {
|
|
43
|
+
throw new Error(`Model ${modelName} not found`);
|
|
44
|
+
}
|
|
45
|
+
return modelInfo.klass;
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"registry.js","sourceRoot":"","sources":["../../src/model/registry.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,0BAA0B,CAAC;AAMlD,MAAM,CAAN,IAAY,SAIX;AAJD,WAAY,SAAS;IACnB,0BAAa,CAAA;IACb,0BAAa,CAAA;IACb,wBAAW,CAAA;AACb,CAAC,EAJW,SAAS,KAAT,SAAS,QAIpB;AAQD,MAAM,cAAc,GAA8B;IAChD,wBAAwB,EAAE;QACxB,IAAI,EAAE,wBAAwB;QAC9B,IAAI,EAAE,SAAS,CAAC,GAAG;QACnB,KAAK,EAAE,MAAM;KACd;IACD,wBAAwB,EAAE;QACxB,IAAI,EAAE,wBAAwB;QAC9B,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,KAAK,EAAE,MAAM;KACd;IACD,+BAA+B,EAAE;QAC/B,IAAI,EAAE,+BAA+B;QACrC,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,KAAK,EAAE,MAAM;KACd;IACD,gBAAgB,EAAE;QAChB,IAAI,EAAE,gBAAgB;QACtB,IAAI,EAAE,SAAS,CAAC,GAAG;QACnB,KAAK,EAAE,MAAM;KACd;IACD,kBAAkB,EAAE;QAClB,IAAI,EAAE,kBAAkB;QACxB,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,KAAK,EAAE,MAAM;KACd;IACD,uBAAuB,EAAE;QACvB,IAAI,EAAE,uBAAuB;QAC7B,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,KAAK,EAAE,MAAM;KACd;CACF,CAAC;AAEF,MAAM,UAAU,eAAe,CAAC,SAAiB;IAC/C,MAAM,SAAS,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;IAE5C,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,SAAS,SAAS,YAAY,CAAC,CAAC;IAClD,CAAC;IAED,OAAO,SAAS,CAAC,KAAK,CAAC;AACzB,CAAC"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { Content } from "../content.js";
|
|
2
|
+
import type { FunctionDeclaration } from "../tools/tool.js";
|
|
3
|
+
export interface LlmRequest {
|
|
4
|
+
model?: string;
|
|
5
|
+
contents: Content[];
|
|
6
|
+
systemInstructions?: string;
|
|
7
|
+
tools?: FunctionDeclaration[];
|
|
8
|
+
thinkingConfig?: ThinkingConfig;
|
|
9
|
+
}
|
|
10
|
+
export interface ThinkingConfig {
|
|
11
|
+
enabled: boolean;
|
|
12
|
+
level?: "low" | "medium" | "high" | "auto";
|
|
13
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"request.js","sourceRoot":"","sources":["../../src/model/request.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { LlmRequest, ThinkingConfig } from "./request.js";
|
|
2
|
+
import type { Content } from "../content.js";
|
|
3
|
+
import type { Tool, FunctionDeclaration } from "../tools/tool.js";
|
|
4
|
+
import { type Skill } from "../skills/skill.js";
|
|
5
|
+
interface BuildLlmRequestOptions {
|
|
6
|
+
agentName: string;
|
|
7
|
+
content: Content;
|
|
8
|
+
historyContent: Content[];
|
|
9
|
+
tools?: Tool[];
|
|
10
|
+
skills?: Skill[];
|
|
11
|
+
description?: string;
|
|
12
|
+
instructions: string;
|
|
13
|
+
thinkingConfig?: ThinkingConfig;
|
|
14
|
+
}
|
|
15
|
+
export declare function buildLlmRequest(options: BuildLlmRequestOptions): LlmRequest;
|
|
16
|
+
export declare function buildTools(tools?: Tool[]): FunctionDeclaration[];
|
|
17
|
+
export declare function buildSkillsTools(skills?: Skill[]): FunctionDeclaration[];
|
|
18
|
+
export declare function buildSystemInstruction(name: string, description: string | undefined, instructions: string): string;
|
|
19
|
+
export {};
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
import { toFunctionDeclaration } from "../skills/skill.js";
|
|
2
|
+
export function buildLlmRequest(options) {
|
|
3
|
+
const { agentName, content, historyContent, tools, skills, description, instructions, thinkingConfig, } = options;
|
|
4
|
+
return {
|
|
5
|
+
contents: mergeAdjacentContents([...historyContent, content]),
|
|
6
|
+
tools: [...buildTools(tools), ...buildSkillsTools(skills)],
|
|
7
|
+
systemInstructions: buildSystemInstruction(agentName, description, instructions),
|
|
8
|
+
thinkingConfig,
|
|
9
|
+
};
|
|
10
|
+
}
|
|
11
|
+
function mergeAdjacentContents(contents) {
|
|
12
|
+
const result = [];
|
|
13
|
+
for (const content of contents) {
|
|
14
|
+
const last = result[result.length - 1];
|
|
15
|
+
if (last && last.role === content.role) {
|
|
16
|
+
last.parts.push(...content.parts);
|
|
17
|
+
}
|
|
18
|
+
else {
|
|
19
|
+
result.push({ ...content, parts: [...content.parts] });
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
return result;
|
|
23
|
+
}
|
|
24
|
+
export function buildTools(tools) {
|
|
25
|
+
if (!tools)
|
|
26
|
+
return [];
|
|
27
|
+
return tools.map((t) => t.toFunctionDeclaration());
|
|
28
|
+
}
|
|
29
|
+
export function buildSkillsTools(skills) {
|
|
30
|
+
if (!skills)
|
|
31
|
+
return [];
|
|
32
|
+
return skills.map((skill) => toFunctionDeclaration(skill));
|
|
33
|
+
}
|
|
34
|
+
export function buildSystemInstruction(name, description, instructions) {
|
|
35
|
+
return [
|
|
36
|
+
`You are an agent. Your internal name is "${name}".`,
|
|
37
|
+
description ? `The description about you is "${description}"` : "",
|
|
38
|
+
instructions,
|
|
39
|
+
]
|
|
40
|
+
.filter(Boolean)
|
|
41
|
+
.join("\n\n");
|
|
42
|
+
}
|
|
43
|
+
//# sourceMappingURL=request_builder_utils.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"request_builder_utils.js","sourceRoot":"","sources":["../../src/model/request_builder_utils.ts"],"names":[],"mappings":"AAGA,OAAO,EAAc,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAavE,MAAM,UAAU,eAAe,CAAC,OAA+B;IAC7D,MAAM,EACJ,SAAS,EACT,OAAO,EACP,cAAc,EACd,KAAK,EACL,MAAM,EACN,WAAW,EACX,YAAY,EACZ,cAAc,GACf,GAAG,OAAO,CAAC;IAEZ,OAAO;QACL,QAAQ,EAAE,qBAAqB,CAAC,CAAC,GAAG,cAAc,EAAE,OAAO,CAAC,CAAC;QAC7D,KAAK,EAAE,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,EAAE,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAC1D,kBAAkB,EAAE,sBAAsB,CACxC,SAAS,EACT,WAAW,EACX,YAAY,CACb;QACD,cAAc;KACf,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAmB;IAChD,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACvC,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,CAAC,IAAI,EAAE,CAAC;YACvC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,KAAc;IACvC,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IAEtB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,qBAAqB,EAAE,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAgB;IAC/C,IAAI,CAAC,MAAM;QAAE,OAAO,EAAE,CAAC;IAEvB,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,IAAY,EACZ,WAA+B,EAC/B,YAAoB;IAEpB,OAAO;QACL,4CAA4C,IAAI,IAAI;QACpD,WAAW,CAAC,CAAC,CAAC,iCAAiC,WAAW,GAAG,CAAC,CAAC,CAAC,EAAE;QAClE,YAAY;KACb;SACE,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,MAAM,CAAC,CAAC;AAClB,CAAC"}
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
import type { Content } from "../content.js";
|
|
2
|
+
export interface UsageMetadata {
|
|
3
|
+
inputTokens?: number;
|
|
4
|
+
outputTokens?: number;
|
|
5
|
+
cachedTokens?: number;
|
|
6
|
+
cost?: {
|
|
7
|
+
amount: number;
|
|
8
|
+
currency?: string;
|
|
9
|
+
};
|
|
10
|
+
}
|
|
11
|
+
export type FinishReason = string;
|
|
12
|
+
/**
|
|
13
|
+
* LLM response class that provides the first candidate response from the
|
|
14
|
+
* model if available. Otherwise, returns error code and message.
|
|
15
|
+
*/
|
|
16
|
+
export interface LlmResponse {
|
|
17
|
+
/**
|
|
18
|
+
* The content of the response.
|
|
19
|
+
*/
|
|
20
|
+
content?: Content;
|
|
21
|
+
/**
|
|
22
|
+
* Indicates whether the text content is part of a unfinished text stream.
|
|
23
|
+
* Only used for streaming mode and when the content is plain text.
|
|
24
|
+
*/
|
|
25
|
+
partial?: boolean;
|
|
26
|
+
/**
|
|
27
|
+
* Indicates whether the response from the model is complete.
|
|
28
|
+
* Only used for streaming mode.
|
|
29
|
+
*/
|
|
30
|
+
turnComplete?: boolean;
|
|
31
|
+
/**
|
|
32
|
+
* Error code if the response is an error. Code varies by model.
|
|
33
|
+
*/
|
|
34
|
+
errorCode?: string;
|
|
35
|
+
/**
|
|
36
|
+
* Error message if the response is an error.
|
|
37
|
+
*/
|
|
38
|
+
errorMessage?: string;
|
|
39
|
+
/**
|
|
40
|
+
* Flag indicating that LLM was interrupted when generating the content.
|
|
41
|
+
* Usually it's due to user interruption during a bidi streaming.
|
|
42
|
+
*/
|
|
43
|
+
interrupted?: boolean;
|
|
44
|
+
/**
|
|
45
|
+
* The custom metadata of the LlmResponse.
|
|
46
|
+
* An optional key-value pair to label an LlmResponse.
|
|
47
|
+
* NOTE: the entire object must be JSON serializable.
|
|
48
|
+
*/
|
|
49
|
+
customMetadata?: {
|
|
50
|
+
[key: string]: unknown;
|
|
51
|
+
};
|
|
52
|
+
/**
|
|
53
|
+
* The finish reason of the response.
|
|
54
|
+
*/
|
|
55
|
+
finishReason?: FinishReason;
|
|
56
|
+
/**
|
|
57
|
+
* The usage metadata of the LlmResponse.
|
|
58
|
+
*/
|
|
59
|
+
usageMetadata?: UsageMetadata;
|
|
60
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"response.js","sourceRoot":"","sources":["../../src/model/response.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { LlmModel } from "./model.js";
|
|
2
|
+
import { type AgentEvent } from "../agent/agent_event.js";
|
|
3
|
+
import type { Content } from "../content.js";
|
|
4
|
+
export declare class UtilLlm {
|
|
5
|
+
private model;
|
|
6
|
+
constructor(model: LlmModel);
|
|
7
|
+
generateSessionTitle(events: AgentEvent[]): Promise<string>;
|
|
8
|
+
compactHistory(contents: Content[]): Promise<string>;
|
|
9
|
+
scanProjectConstants(context: string): Promise<Record<string, any>>;
|
|
10
|
+
}
|
|
@@ -0,0 +1,149 @@
|
|
|
1
|
+
import { isMessageEvent } from "../agent/agent_event.js";
|
|
2
|
+
import { logger } from "../logger.js";
|
|
3
|
+
export class UtilLlm {
|
|
4
|
+
constructor(model) {
|
|
5
|
+
this.model = model;
|
|
6
|
+
}
|
|
7
|
+
async generateSessionTitle(events) {
|
|
8
|
+
const conversationText = events
|
|
9
|
+
.filter(isMessageEvent)
|
|
10
|
+
.map((event) => {
|
|
11
|
+
const role = event.role === "user" ? "User" : "Agent";
|
|
12
|
+
const text = event.parts
|
|
13
|
+
.filter((part) => part.type === "text")
|
|
14
|
+
.map((part) => part.text)
|
|
15
|
+
.join(" ");
|
|
16
|
+
return `[${role}]: ${text}`;
|
|
17
|
+
})
|
|
18
|
+
.join("\n");
|
|
19
|
+
if (!conversationText) {
|
|
20
|
+
return "New Session";
|
|
21
|
+
}
|
|
22
|
+
const prompt = `Based on the following conversation, generate a short, user-friendly title (3-5 words). Do not use quotes or special characters in the title.\n\nConversation:\n${conversationText}\n\nTitle:`;
|
|
23
|
+
const request = {
|
|
24
|
+
contents: [
|
|
25
|
+
{
|
|
26
|
+
role: "user",
|
|
27
|
+
parts: [{ type: "text", text: prompt }],
|
|
28
|
+
},
|
|
29
|
+
],
|
|
30
|
+
systemInstructions: "You are a helpful assistant that generates concise titles for conversations.",
|
|
31
|
+
};
|
|
32
|
+
try {
|
|
33
|
+
const generator = this.model.run(request);
|
|
34
|
+
let title = "";
|
|
35
|
+
for await (const response of generator) {
|
|
36
|
+
if (response.errorMessage) {
|
|
37
|
+
throw new Error(response.errorMessage);
|
|
38
|
+
}
|
|
39
|
+
if (response.content && response.content.parts) {
|
|
40
|
+
for (const part of response.content.parts) {
|
|
41
|
+
if (part.type === "text") {
|
|
42
|
+
title += part.text;
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
return title.trim() || "New Session";
|
|
48
|
+
}
|
|
49
|
+
catch (e) {
|
|
50
|
+
console.error("Failed to generate session title:", e);
|
|
51
|
+
return "New Session";
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
async compactHistory(contents) {
|
|
55
|
+
const conversationText = contents
|
|
56
|
+
.map((content) => {
|
|
57
|
+
const role = content.role === "user" ? "User" : "Agent";
|
|
58
|
+
const text = content.parts
|
|
59
|
+
.filter((part) => part.type === "text")
|
|
60
|
+
.map((part) => part.text)
|
|
61
|
+
.join(" ");
|
|
62
|
+
return `[${role}]: ${text}`;
|
|
63
|
+
})
|
|
64
|
+
.join("\n");
|
|
65
|
+
const prompt = `Summarize the following conversation history strictly retaining facts, code snippets, and user preferences. Be concise.\n\nConversation:\n${conversationText}\n\nSummary:`;
|
|
66
|
+
const request = {
|
|
67
|
+
contents: [
|
|
68
|
+
{
|
|
69
|
+
role: "user",
|
|
70
|
+
parts: [{ type: "text", text: prompt }],
|
|
71
|
+
},
|
|
72
|
+
],
|
|
73
|
+
systemInstructions: "You are a helpful assistant that summarizes conversation history.",
|
|
74
|
+
};
|
|
75
|
+
try {
|
|
76
|
+
const generator = this.model.run(request);
|
|
77
|
+
let summary = "";
|
|
78
|
+
for await (const response of generator) {
|
|
79
|
+
if (response.errorMessage) {
|
|
80
|
+
throw new Error(response.errorMessage);
|
|
81
|
+
}
|
|
82
|
+
if (response.content && response.content.parts) {
|
|
83
|
+
for (const part of response.content.parts) {
|
|
84
|
+
if (part.type === "text") {
|
|
85
|
+
summary += part.text;
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
return summary.trim();
|
|
91
|
+
}
|
|
92
|
+
catch (e) {
|
|
93
|
+
logger.error("Failed to compact history:", e);
|
|
94
|
+
throw e;
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
async scanProjectConstants(context) {
|
|
98
|
+
const prompt = `Analyze the following project information and define project constants.
|
|
99
|
+
Identify:
|
|
100
|
+
- language, framework, package managers, build tools
|
|
101
|
+
- project structure
|
|
102
|
+
- coding style, rules and conventions
|
|
103
|
+
|
|
104
|
+
Output the result as a JSON object. Do not include markdown formatting in the response, just the raw JSON.
|
|
105
|
+
|
|
106
|
+
Project Info:
|
|
107
|
+
${context}
|
|
108
|
+
`;
|
|
109
|
+
const request = {
|
|
110
|
+
contents: [
|
|
111
|
+
{
|
|
112
|
+
role: "user",
|
|
113
|
+
parts: [{ type: "text", text: prompt }],
|
|
114
|
+
},
|
|
115
|
+
],
|
|
116
|
+
systemInstructions: "You are a helpful assistant that analyzes codebases and extracts project conventions and constants in JSON format.",
|
|
117
|
+
};
|
|
118
|
+
try {
|
|
119
|
+
const generator = this.model.run(request);
|
|
120
|
+
let resultStr = "";
|
|
121
|
+
for await (const response of generator) {
|
|
122
|
+
if (response.errorMessage) {
|
|
123
|
+
throw new Error(response.errorMessage);
|
|
124
|
+
}
|
|
125
|
+
if (response.content && response.content.parts) {
|
|
126
|
+
for (const part of response.content.parts) {
|
|
127
|
+
if (part.type === "text") {
|
|
128
|
+
resultStr += part.text;
|
|
129
|
+
}
|
|
130
|
+
}
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
let cleanedStr = resultStr.trim();
|
|
134
|
+
if (cleanedStr.startsWith("\`\`\`json")) {
|
|
135
|
+
cleanedStr = cleanedStr.substring(7);
|
|
136
|
+
}
|
|
137
|
+
if (cleanedStr.endsWith("\`\`\`")) {
|
|
138
|
+
cleanedStr = cleanedStr.substring(0, cleanedStr.length - 3);
|
|
139
|
+
}
|
|
140
|
+
cleanedStr = cleanedStr.trim();
|
|
141
|
+
return JSON.parse(cleanedStr);
|
|
142
|
+
}
|
|
143
|
+
catch (e) {
|
|
144
|
+
logger.error("Failed to scan project constants:", e);
|
|
145
|
+
throw e;
|
|
146
|
+
}
|
|
147
|
+
}
|
|
148
|
+
}
|
|
149
|
+
//# sourceMappingURL=util_llm.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"util_llm.js","sourceRoot":"","sources":["../../src/model/util_llm.ts"],"names":[],"mappings":"AACA,OAAO,EAAmB,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAE1E,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,MAAM,OAAO,OAAO;IAGlB,YAAY,KAAe;QACzB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,MAAoB;QAC7C,MAAM,gBAAgB,GAAG,MAAM;aAC5B,MAAM,CAAC,cAAc,CAAC;aACtB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACb,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC;YACtD,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK;iBACrB,MAAM,CAAC,CAAC,IAAI,EAA2B,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;iBAC/D,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;iBACxB,IAAI,CAAC,GAAG,CAAC,CAAC;YACb,OAAO,IAAI,IAAI,MAAM,IAAI,EAAE,CAAC;QAC9B,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,OAAO,aAAa,CAAC;QACvB,CAAC;QAED,MAAM,MAAM,GAAG,mKAAmK,gBAAgB,YAAY,CAAC;QAE/M,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAe;oBACrB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;iBACjD;aACF;YACD,kBAAkB,EAChB,8EAA8E;SACjF,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC1C,IAAI,KAAK,GAAG,EAAE,CAAC;YACf,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBACvC,IAAI,QAAQ,CAAC,YAAY,EAAE,CAAC;oBAC1B,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBACzC,CAAC;gBACD,IAAI,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;oBAC/C,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;wBAC1C,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BACzB,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC;wBACrB,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,KAAK,CAAC,IAAI,EAAE,IAAI,aAAa,CAAC;QACvC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,CAAC,CAAC,CAAC;YACtD,OAAO,aAAa,CAAC;QACvB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAmB;QACtC,MAAM,gBAAgB,GAAG,QAAQ;aAC9B,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;YACf,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC;YACxD,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK;iBACvB,MAAM,CAAC,CAAC,IAAI,EAA2B,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;iBAC/D,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;iBACxB,IAAI,CAAC,GAAG,CAAC,CAAC;YACb,OAAO,IAAI,IAAI,MAAM,IAAI,EAAE,CAAC;QAC9B,CAAC,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,MAAM,GAAG,6IAA6I,gBAAgB,cAAc,CAAC;QAE3L,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAe;oBACrB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;iBACjD;aACF;YACD,kBAAkB,EAChB,mEAAmE;SACtE,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC1C,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBACvC,IAAI,QAAQ,CAAC,YAAY,EAAE,CAAC;oBAC1B,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBACzC,CAAC;gBACD,IAAI,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;oBAC/C,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;wBAC1C,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BACzB,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC;wBACvB,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;QACxB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,OAAe;QACxC,MAAM,MAAM,GAAG;;;;;;;;;EASjB,OAAO;CACR,CAAC;QAEE,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAe;oBACrB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;iBACjD;aACF;YACD,kBAAkB,EAChB,oHAAoH;SACvH,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC1C,IAAI,SAAS,GAAG,EAAE,CAAC;YACnB,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBACvC,IAAI,QAAQ,CAAC,YAAY,EAAE,CAAC;oBAC1B,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBACzC,CAAC;gBACD,IAAI,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;oBAC/C,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;wBAC1C,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BACzB,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC;wBACzB,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,UAAU,GAAG,SAAS,CAAC,IAAI,EAAE,CAAC;YAClC,IAAI,UAAU,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;gBACxC,UAAU,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACvC,CAAC;YACD,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAClC,UAAU,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC9D,CAAC;YACD,UAAU,GAAG,UAAU,CAAC,IAAI,EAAE,CAAC;YAE/B,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE,CAAC,CAAC,CAAC;YACrD,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import { type AgentEvent } from "../agent/agent_event.js";
|
|
2
|
+
export interface SessionMetadata {
|
|
3
|
+
id: string;
|
|
4
|
+
title?: string;
|
|
5
|
+
agentName: string;
|
|
6
|
+
timestamp: string;
|
|
7
|
+
}
|
|
8
|
+
export interface Session extends SessionMetadata {
|
|
9
|
+
id: string;
|
|
10
|
+
title?: string;
|
|
11
|
+
events: AgentEvent[];
|
|
12
|
+
agentName: string;
|
|
13
|
+
timestamp: string;
|
|
14
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.js","sourceRoot":"","sources":["../../src/session/session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmB,MAAM,yBAAyB,CAAC"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import type { Session, SessionMetadata } from "./session.js";
|
|
2
|
+
import { type AgentEvent } from "../agent/agent_event.js";
|
|
3
|
+
export declare class SessionFileService {
|
|
4
|
+
private rootDir;
|
|
5
|
+
private initialized;
|
|
6
|
+
private locks;
|
|
7
|
+
constructor();
|
|
8
|
+
private init;
|
|
9
|
+
private lock;
|
|
10
|
+
getSession(sessionId: string): Promise<Session>;
|
|
11
|
+
getSessionMetadata(sessionId: string): Promise<SessionMetadata | undefined>;
|
|
12
|
+
createSession(agentName: string, events: AgentEvent[]): Promise<Session>;
|
|
13
|
+
updateSession(sessionId: string, { title }: {
|
|
14
|
+
title: string;
|
|
15
|
+
}): Promise<void>;
|
|
16
|
+
appendEvent(sessionId: string, agentEvent: AgentEvent): Promise<void>;
|
|
17
|
+
listSessions(): Promise<Array<SessionMetadata>>;
|
|
18
|
+
}
|
|
19
|
+
export declare function listFiles(folderPath: string): Promise<string[]>;
|
|
20
|
+
export declare function getSessionFileName(rootDir: string, sessionId: string): string;
|
|
21
|
+
export declare function getSessionMetadataFileName(rootDir: string, sessionId: string): string;
|
|
22
|
+
export declare function saveToFile<T>(filePath: string, data: T): Promise<void>;
|
|
23
|
+
export declare function loadFileData<T>(filePath: string): Promise<T | undefined>;
|