atabey 0.0.12 → 0.0.13

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.
Files changed (232) hide show
  1. package/ATABEY.md +45 -499
  2. package/PRIVACY.md +161 -0
  3. package/README.md +353 -88
  4. package/bin/validate-al-registry.js +4 -4
  5. package/dist/framework-mcp/src/tools/definitions.js +23 -1
  6. package/dist/framework-mcp/src/tools/definitions.js.map +1 -1
  7. package/dist/framework-mcp/src/tools/gateway/llm_gateway_query.d.ts +19 -0
  8. package/dist/framework-mcp/src/tools/gateway/llm_gateway_query.js +138 -0
  9. package/dist/framework-mcp/src/tools/gateway/llm_gateway_query.js.map +1 -0
  10. package/dist/framework-mcp/src/tools/gateway/llm_gateway_status.d.ts +8 -0
  11. package/dist/framework-mcp/src/tools/gateway/llm_gateway_status.js +91 -0
  12. package/dist/framework-mcp/src/tools/gateway/llm_gateway_status.js.map +1 -0
  13. package/dist/framework-mcp/src/tools/index.d.ts +1 -1
  14. package/dist/framework-mcp/src/tools/index.js +35 -20
  15. package/dist/framework-mcp/src/tools/index.js.map +1 -1
  16. package/dist/framework-mcp/src/tools/quality/analyze_code_quality.d.ts +11 -0
  17. package/dist/framework-mcp/src/tools/quality/analyze_code_quality.js +166 -0
  18. package/dist/framework-mcp/src/tools/quality/analyze_code_quality.js.map +1 -0
  19. package/dist/framework-mcp/src/tools/quality/check_architecture_compliance.d.ts +11 -0
  20. package/dist/framework-mcp/src/tools/quality/check_architecture_compliance.js +135 -0
  21. package/dist/framework-mcp/src/tools/quality/check_architecture_compliance.js.map +1 -0
  22. package/dist/framework-mcp/src/tools/schemas.d.ts +88 -8
  23. package/dist/framework-mcp/src/tools/schemas.js +29 -0
  24. package/dist/framework-mcp/src/tools/schemas.js.map +1 -1
  25. package/dist/framework-mcp/src/tools/shell/run_command.js +2 -2
  26. package/dist/framework-mcp/src/tools/shell/run_command.js.map +1 -1
  27. package/dist/src/cli/commands/check.js +2 -2
  28. package/dist/src/cli/commands/check.js.map +1 -1
  29. package/dist/src/cli/commands/dashboard.d.ts +1 -1
  30. package/dist/src/cli/commands/dashboard.js +264 -41
  31. package/dist/src/cli/commands/dashboard.js.map +1 -1
  32. package/dist/src/cli/commands/gateway.d.ts +1 -0
  33. package/dist/src/cli/commands/gateway.js +172 -0
  34. package/dist/src/cli/commands/gateway.js.map +1 -0
  35. package/dist/src/cli/commands/init/create-agent.js +3 -3
  36. package/dist/src/cli/commands/init/create-agent.js.map +1 -1
  37. package/dist/src/cli/commands/init/scaffold-core.d.ts +1 -1
  38. package/dist/src/cli/commands/init/scaffold-core.js +1 -1
  39. package/dist/src/cli/commands/init/scaffold-core.js.map +1 -1
  40. package/dist/src/cli/commands/init.d.ts +2 -0
  41. package/dist/src/cli/commands/init.js +105 -23
  42. package/dist/src/cli/commands/init.js.map +1 -1
  43. package/dist/src/cli/commands/mcp.d.ts +7 -0
  44. package/dist/src/cli/commands/mcp.js +45 -22
  45. package/dist/src/cli/commands/mcp.js.map +1 -1
  46. package/dist/src/cli/commands/orchestrate.d.ts +8 -7
  47. package/dist/src/cli/commands/orchestrate.js +2 -1
  48. package/dist/src/cli/commands/orchestrate.js.map +1 -1
  49. package/dist/src/cli/commands/quickstart.d.ts +5 -5
  50. package/dist/src/cli/commands/quickstart.js +36 -36
  51. package/dist/src/cli/commands/quickstart.js.map +1 -1
  52. package/dist/src/cli/index.js +230 -320
  53. package/dist/src/cli/index.js.map +1 -1
  54. package/dist/src/cli/platforms/core.d.ts +12 -0
  55. package/dist/src/cli/platforms/core.js +73 -0
  56. package/dist/src/cli/platforms/core.js.map +1 -0
  57. package/dist/src/cli/platforms/index.d.ts +5 -0
  58. package/dist/src/cli/platforms/index.js +6 -0
  59. package/dist/src/cli/platforms/index.js.map +1 -0
  60. package/dist/src/cli/platforms/paths.d.ts +15 -0
  61. package/dist/src/cli/platforms/paths.js +102 -0
  62. package/dist/src/cli/platforms/paths.js.map +1 -0
  63. package/dist/src/cli/platforms/scaffold.d.ts +2 -0
  64. package/dist/src/cli/platforms/scaffold.js +73 -0
  65. package/dist/src/cli/platforms/scaffold.js.map +1 -0
  66. package/dist/src/cli/platforms/utils.d.ts +4 -0
  67. package/dist/src/cli/platforms/utils.js +76 -0
  68. package/dist/src/cli/platforms/utils.js.map +1 -0
  69. package/dist/src/cli/shims.js +4 -4
  70. package/dist/src/cli/utils/fs.d.ts +1 -1
  71. package/dist/src/cli/utils/fs.js +1 -1
  72. package/dist/src/cli/utils/fs.js.map +1 -1
  73. package/dist/src/modules/agents/definitions.d.ts +2 -2
  74. package/dist/src/modules/agents/definitions.js +59 -23
  75. package/dist/src/modules/agents/definitions.js.map +1 -1
  76. package/dist/src/modules/contracts/api.d.ts +178 -0
  77. package/dist/src/modules/contracts/api.js +56 -0
  78. package/dist/src/modules/contracts/api.js.map +1 -0
  79. package/dist/src/modules/contracts/index.d.ts +5 -0
  80. package/dist/src/modules/contracts/index.js +6 -0
  81. package/dist/src/modules/contracts/index.js.map +1 -0
  82. package/dist/src/modules/engines/agent-executor.d.ts +27 -28
  83. package/dist/src/modules/engines/agent-executor.js +97 -69
  84. package/dist/src/modules/engines/agent-executor.js.map +1 -1
  85. package/dist/src/modules/engines/quality-gate.d.ts +7 -7
  86. package/dist/src/modules/engines/quality-gate.js +7 -7
  87. package/dist/src/modules/engines/routing-engine.d.ts +18 -27
  88. package/dist/src/modules/engines/routing-engine.js +70 -44
  89. package/dist/src/modules/engines/routing-engine.js.map +1 -1
  90. package/dist/src/modules/gateway/circuit-breaker.d.ts +40 -0
  91. package/dist/src/modules/gateway/circuit-breaker.js +163 -0
  92. package/dist/src/modules/gateway/circuit-breaker.js.map +1 -0
  93. package/dist/src/modules/gateway/cost-tracker.d.ts +55 -0
  94. package/dist/src/modules/gateway/cost-tracker.js +127 -0
  95. package/dist/src/modules/gateway/cost-tracker.js.map +1 -0
  96. package/dist/src/modules/gateway/gateway.d.ts +68 -0
  97. package/dist/src/modules/gateway/gateway.js +235 -0
  98. package/dist/src/modules/gateway/gateway.js.map +1 -0
  99. package/dist/src/modules/gateway/index.d.ts +19 -0
  100. package/dist/src/modules/gateway/index.js +19 -0
  101. package/dist/src/modules/gateway/index.js.map +1 -0
  102. package/dist/src/modules/gateway/load-balancer.d.ts +86 -0
  103. package/dist/src/modules/gateway/load-balancer.js +230 -0
  104. package/dist/src/modules/gateway/load-balancer.js.map +1 -0
  105. package/dist/src/modules/gateway/providers/anthropic.d.ts +14 -0
  106. package/dist/src/modules/gateway/providers/anthropic.js +169 -0
  107. package/dist/src/modules/gateway/providers/anthropic.js.map +1 -0
  108. package/dist/src/modules/gateway/providers/base.d.ts +38 -0
  109. package/dist/src/modules/gateway/providers/base.js +42 -0
  110. package/dist/src/modules/gateway/providers/base.js.map +1 -0
  111. package/dist/src/modules/gateway/providers/google.d.ts +18 -0
  112. package/dist/src/modules/gateway/providers/google.js +176 -0
  113. package/dist/src/modules/gateway/providers/google.js.map +1 -0
  114. package/dist/src/modules/gateway/providers/grok.d.ts +14 -0
  115. package/dist/src/modules/gateway/providers/grok.js +166 -0
  116. package/dist/src/modules/gateway/providers/grok.js.map +1 -0
  117. package/dist/src/modules/gateway/providers/index.d.ts +21 -0
  118. package/dist/src/modules/gateway/providers/index.js +38 -0
  119. package/dist/src/modules/gateway/providers/index.js.map +1 -0
  120. package/dist/src/modules/gateway/providers/ollama.d.ts +23 -0
  121. package/dist/src/modules/gateway/providers/ollama.js +187 -0
  122. package/dist/src/modules/gateway/providers/ollama.js.map +1 -0
  123. package/dist/src/modules/gateway/providers/openai.d.ts +14 -0
  124. package/dist/src/modules/gateway/providers/openai.js +170 -0
  125. package/dist/src/modules/gateway/providers/openai.js.map +1 -0
  126. package/dist/src/modules/gateway/rate-limiter.d.ts +41 -0
  127. package/dist/src/modules/gateway/rate-limiter.js +119 -0
  128. package/dist/src/modules/gateway/rate-limiter.js.map +1 -0
  129. package/dist/src/modules/gateway/types.d.ts +114 -0
  130. package/dist/src/modules/gateway/types.js +8 -0
  131. package/dist/src/modules/gateway/types.js.map +1 -0
  132. package/dist/src/modules/memory/core.d.ts +4 -12
  133. package/dist/src/modules/memory/core.js +10 -17
  134. package/dist/src/modules/memory/core.js.map +1 -1
  135. package/dist/src/modules/memory/embedding.d.ts +28 -0
  136. package/dist/src/modules/memory/embedding.js +131 -0
  137. package/dist/src/modules/memory/embedding.js.map +1 -0
  138. package/dist/src/modules/providers/definitions.d.ts +9 -0
  139. package/dist/src/modules/providers/definitions.js +141 -0
  140. package/dist/src/modules/providers/definitions.js.map +1 -0
  141. package/dist/src/modules/providers/registry.d.ts +12 -0
  142. package/dist/src/modules/providers/registry.js +19 -0
  143. package/dist/src/modules/providers/registry.js.map +1 -0
  144. package/dist/src/modules/providers/shared.d.ts +1 -0
  145. package/dist/src/modules/providers/shared.js +105 -0
  146. package/dist/src/modules/providers/shared.js.map +1 -0
  147. package/dist/src/modules/providers/types.d.ts +14 -0
  148. package/dist/src/modules/providers/types.js +2 -0
  149. package/dist/src/modules/providers/types.js.map +1 -0
  150. package/dist/src/modules/skills/adapter-skills.d.ts +68 -0
  151. package/dist/src/modules/skills/adapter-skills.js +450 -0
  152. package/dist/src/modules/skills/adapter-skills.js.map +1 -0
  153. package/dist/src/shared/audit.d.ts +105 -0
  154. package/dist/src/shared/audit.js +220 -0
  155. package/dist/src/shared/audit.js.map +1 -0
  156. package/dist/src/shared/fs.js +4 -2
  157. package/dist/src/shared/fs.js.map +1 -1
  158. package/dist/src/shared/lock.d.ts +82 -0
  159. package/dist/src/shared/lock.js +160 -0
  160. package/dist/src/shared/lock.js.map +1 -0
  161. package/dist/src/shared/logger.d.ts +2 -0
  162. package/dist/src/shared/logger.js +15 -5
  163. package/dist/src/shared/logger.js.map +1 -1
  164. package/dist/src/shared/pii.d.ts +37 -0
  165. package/dist/src/shared/pii.js +122 -0
  166. package/dist/src/shared/pii.js.map +1 -0
  167. package/dist/src/shared/retention.d.ts +63 -0
  168. package/dist/src/shared/retention.js +171 -0
  169. package/dist/src/shared/retention.js.map +1 -0
  170. package/dist/src/shared/storage.d.ts +3 -1
  171. package/dist/src/shared/storage.js +3 -1
  172. package/dist/src/shared/storage.js.map +1 -1
  173. package/dist/tests/adapter.test.js +1 -1
  174. package/dist/tests/adapter.test.js.map +1 -1
  175. package/dist/tests/adapters/paths.test.js +4 -4
  176. package/dist/tests/adapters/paths.test.js.map +1 -1
  177. package/dist/tests/cli/commands/dashboard-coverage.test.d.ts +1 -0
  178. package/dist/tests/cli/commands/dashboard-coverage.test.js +88 -0
  179. package/dist/tests/cli/commands/dashboard-coverage.test.js.map +1 -0
  180. package/dist/tests/cli/init/adapter_init.test.js +4 -5
  181. package/dist/tests/cli/init/adapter_init.test.js.map +1 -1
  182. package/dist/tests/cli/init/unified_init.test.js +5 -7
  183. package/dist/tests/cli/init/unified_init.test.js.map +1 -1
  184. package/dist/tests/integration/agent_flow.test.js +14 -8
  185. package/dist/tests/integration/agent_flow.test.js.map +1 -1
  186. package/dist/tests/modules/adapters/adapters.test.js +2 -2
  187. package/dist/tests/modules/adapters/adapters.test.js.map +1 -1
  188. package/dist/tests/modules/contracts/contracts.test.d.ts +1 -0
  189. package/dist/tests/modules/contracts/contracts.test.js +73 -0
  190. package/dist/tests/modules/contracts/contracts.test.js.map +1 -0
  191. package/dist/tests/modules/engines/agent-executor.test.js +30 -23
  192. package/dist/tests/modules/engines/agent-executor.test.js.map +1 -1
  193. package/dist/tests/modules/gateway/circuit-breaker.test.d.ts +1 -0
  194. package/dist/tests/modules/gateway/circuit-breaker.test.js +124 -0
  195. package/dist/tests/modules/gateway/circuit-breaker.test.js.map +1 -0
  196. package/dist/tests/modules/gateway/cost-tracker.test.d.ts +1 -0
  197. package/dist/tests/modules/gateway/cost-tracker.test.js +74 -0
  198. package/dist/tests/modules/gateway/cost-tracker.test.js.map +1 -0
  199. package/dist/tests/modules/gateway/load-balancer.test.d.ts +1 -0
  200. package/dist/tests/modules/gateway/load-balancer.test.js +70 -0
  201. package/dist/tests/modules/gateway/load-balancer.test.js.map +1 -0
  202. package/dist/tests/modules/gateway/rate-limiter.test.d.ts +1 -0
  203. package/dist/tests/modules/gateway/rate-limiter.test.js +62 -0
  204. package/dist/tests/modules/gateway/rate-limiter.test.js.map +1 -0
  205. package/dist/tests/orchestrator-dependencies.test.js +23 -6
  206. package/dist/tests/orchestrator-dependencies.test.js.map +1 -1
  207. package/dist/tests/shared/audit.test.d.ts +1 -0
  208. package/dist/tests/shared/audit.test.js +53 -0
  209. package/dist/tests/shared/audit.test.js.map +1 -0
  210. package/dist/tests/storage-operations.test.d.ts +1 -0
  211. package/dist/tests/storage-operations.test.js +116 -0
  212. package/dist/tests/storage-operations.test.js.map +1 -0
  213. package/framework-mcp/README.md +324 -77
  214. package/framework-mcp/dist/dashboard/assets/index-BAPGLiWE.js +267 -0
  215. package/framework-mcp/dist/dashboard/index.html +15 -0
  216. package/framework-mcp/dist/framework-mcp/src/tools/definitions.js +23 -1
  217. package/framework-mcp/dist/framework-mcp/src/tools/gateway/llm_gateway_query.js +137 -0
  218. package/framework-mcp/dist/framework-mcp/src/tools/gateway/llm_gateway_status.js +90 -0
  219. package/framework-mcp/dist/framework-mcp/src/tools/index.js +35 -20
  220. package/framework-mcp/dist/framework-mcp/src/tools/quality/analyze_code_quality.js +165 -0
  221. package/framework-mcp/dist/framework-mcp/src/tools/quality/check_architecture_compliance.js +134 -0
  222. package/framework-mcp/dist/framework-mcp/src/tools/schemas.js +29 -0
  223. package/framework-mcp/dist/framework-mcp/src/tools/shell/run_command.js +2 -2
  224. package/framework-mcp/dist/src/modules/memory/core.js +10 -17
  225. package/framework-mcp/dist/src/modules/memory/embedding.js +130 -0
  226. package/framework-mcp/dist/src/shared/fs.js +4 -2
  227. package/framework-mcp/dist/src/shared/logger.js +15 -5
  228. package/framework-mcp/dist/src/shared/pii.js +121 -0
  229. package/framework-mcp/dist/src/shared/storage.js +3 -1
  230. package/framework-mcp/package.json +1 -1
  231. package/package.json +10 -4
  232. package/templates/full/ATABEY_FULL.md +515 -0
@@ -0,0 +1 @@
1
+ {"version":3,"file":"grok.js","sourceRoot":"","sources":["../../../../../src/modules/gateway/providers/grok.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,MAAM,OAAO,YAAa,SAAQ,YAAY;IAC1C,YAAY,MAAgC;QACxC,KAAK,CAAC;YACF,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,qBAAqB;YAC9B,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,EAAE;YACrC,MAAM,EAAE;gBACJ,QAAQ,EAAE,aAAa,EAAE,WAAW;aACvC;YACD,YAAY,EAAE,QAAQ;YACtB,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,GAAG;YACb,QAAQ,EAAE,KAAK;YACf,oBAAoB,EAAE,KAAK;YAC3B,qBAAqB,EAAE,KAAK;YAC5B,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,CAAC;YACX,MAAM,EAAE,CAAC;YACT,GAAG,MAAM;SACZ,CAAC,CAAC;IACP,CAAC;IAED,IAAI,EAAE,KAAa,OAAO,MAAM,CAAC,CAAC,CAAC;IACnC,IAAI,IAAI,KAAa,OAAO,UAAU,CAAC,CAAC,CAAC;IAEzC,KAAK,CAAC,IAAI,CAAC,OAAuB;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QACxD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAEhC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,aAAa,CAAC,sDAAsD,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACtG,CAAC;QAED,IAAI,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAE5E,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,mBAAmB,EAAE;gBACpE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;iBAChD;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK;oBACL,QAAQ,EAAE;wBACN,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBACjF,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BAC1B,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,OAAO,EAAE,CAAC,CAAC,OAAO;yBACrB,CAAC,CAAC;qBACN;oBACD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;oBACvC,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;oBACrC,MAAM,EAAE,KAAK;iBAChB,CAAC;gBACF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC5B,CAAC,CAAC;YAEH,YAAY,CAAC,OAAO,CAAC,CAAC;YAEtB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,aAAa,CACrB,kBAAkB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,EAClD,KAAK,EACL,OAAO,CACV,CAAC;YACN,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAIrC,CAAC;YAEI,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACzC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;YAC1D,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YACrH,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,EAAE,iBAAiB,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YACnF,MAAM,IAAI,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB;gBAC7D,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC;YAEjE,OAAO;gBACH,OAAO,EAAE,IAAI;gBACb,OAAO;gBACP,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,KAAK;gBAC1B,QAAQ,EAAE,MAAM;gBAChB,KAAK,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,GAAG,YAAY,EAAE;gBAC7E,SAAS;gBACT,IAAI;gBACJ,MAAM,EAAE,KAAK;gBACb,OAAO;aACV,CAAC;QACN,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACxD,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACpE,CAAC;YACD,OAAO,IAAI,CAAC,aAAa,CACrB,uBAAwB,KAAe,CAAC,OAAO,EAAE,EACjD,KAAK,EACL,OAAO,CACV,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,OAAuB;QACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QAExD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,MAAM,EAAE,OAAO,EAAE,6BAA6B,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;YACtF,OAAO;QACX,CAAC;QAED,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,mBAAmB,EAAE;gBACpE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;iBAChD;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK;oBACL,QAAQ,EAAE;wBACN,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBACjF,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BAC1B,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,OAAO,EAAE,CAAC,CAAC,OAAO;yBACrB,CAAC,CAAC;qBACN;oBACD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;oBACvC,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;oBACrC,MAAM,EAAE,IAAI;iBACf,CAAC;aACL,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,EAAE,OAAO,EAAE,qBAAqB,QAAQ,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;gBAC/F,OAAO;YACX,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACV,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;gBAC3E,OAAO;YACX,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,OAAO,IAAI,EAAE,CAAC;gBACV,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACvB,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAClC,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;4BACpB,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;4BAC3D,OAAO;wBACX,CAAC;wBACD,IAAI,CAAC;4BACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAE3C,CAAC;4BACY,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;4BAC1D,IAAI,OAAO,EAAE,CAAC;gCACV,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;4BAC5D,CAAC;wBACL,CAAC;wBAAC,MAAM,CAAC;4BACL,iBAAiB;wBACrB,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;YAED,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QAC/D,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,MAAM;gBACF,OAAO,EAAE,sBAAuB,KAAe,CAAC,OAAO,EAAE;gBACzD,IAAI,EAAE,IAAI;gBACV,QAAQ,EAAE,MAAM;gBAChB,KAAK;aACR,CAAC;QACN,CAAC;IACL,CAAC;CACJ"}
@@ -0,0 +1,21 @@
1
+ /**
2
+ * ATABEY LLM Gateway — Provider Index
3
+ *
4
+ * Exports all provider implementations and a factory function.
5
+ */
6
+ import { ProviderConfig, ProviderId } from "../types.js";
7
+ import { BaseProvider } from "./base.js";
8
+ export { AnthropicProvider } from "./anthropic.js";
9
+ export { BaseProvider } from "./base.js";
10
+ export { GoogleProvider } from "./google.js";
11
+ export { GrokProvider } from "./grok.js";
12
+ export { OllamaProvider } from "./ollama.js";
13
+ export { OpenAIProvider } from "./openai.js";
14
+ /**
15
+ * Create a provider instance by ID with optional config overrides.
16
+ */
17
+ export declare function createProvider(id: ProviderId, config?: Partial<ProviderConfig>): BaseProvider;
18
+ /**
19
+ * Get all available provider IDs.
20
+ */
21
+ export declare function getAvailableProviderIds(): ProviderId[];
@@ -0,0 +1,38 @@
1
+ import { AnthropicProvider } from "./anthropic.js";
2
+ import { GoogleProvider } from "./google.js";
3
+ import { GrokProvider } from "./grok.js";
4
+ import { OllamaProvider } from "./ollama.js";
5
+ import { OpenAIProvider } from "./openai.js";
6
+ export { AnthropicProvider } from "./anthropic.js";
7
+ export { BaseProvider } from "./base.js";
8
+ export { GoogleProvider } from "./google.js";
9
+ export { GrokProvider } from "./grok.js";
10
+ export { OllamaProvider } from "./ollama.js";
11
+ export { OpenAIProvider } from "./openai.js";
12
+ /**
13
+ * Provider class map for dynamic instantiation.
14
+ */
15
+ const PROVIDER_CLASSES = {
16
+ openai: OpenAIProvider,
17
+ anthropic: AnthropicProvider,
18
+ google: GoogleProvider,
19
+ grok: GrokProvider,
20
+ ollama: OllamaProvider,
21
+ };
22
+ /**
23
+ * Create a provider instance by ID with optional config overrides.
24
+ */
25
+ export function createProvider(id, config) {
26
+ const ProviderClass = PROVIDER_CLASSES[id];
27
+ if (!ProviderClass) {
28
+ throw new Error(`Unknown provider: ${id}. Available: ${Object.keys(PROVIDER_CLASSES).join(", ")}`);
29
+ }
30
+ return new ProviderClass(config);
31
+ }
32
+ /**
33
+ * Get all available provider IDs.
34
+ */
35
+ export function getAvailableProviderIds() {
36
+ return Object.keys(PROVIDER_CLASSES);
37
+ }
38
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/modules/gateway/providers/index.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAEnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAK7C;;GAEG;AACH,MAAM,gBAAgB,GAA4C;IAC9D,MAAM,EAAE,cAAc;IACtB,SAAS,EAAE,iBAAiB;IAC5B,MAAM,EAAE,cAAc;IACtB,IAAI,EAAE,YAAY;IAClB,MAAM,EAAE,cAAc;CACzB,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,EAAc,EAAE,MAAgC;IAC3E,MAAM,aAAa,GAAG,gBAAgB,CAAC,EAAE,CAAC,CAAC;IAC3C,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,qBAAqB,EAAE,gBAAgB,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACvG,CAAC;IAED,OAAO,IAAK,aAAqC,CAAC,MAAM,CAAC,CAAC;AAC9D,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB;IACnC,OAAO,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAiB,CAAC;AACzD,CAAC"}
@@ -0,0 +1,23 @@
1
+ /**
2
+ * ATABEY LLM Gateway — Ollama (Local LLM) Provider
3
+ *
4
+ * Supports local models via Ollama API (llama3, mistral, codellama, etc.)
5
+ * Defaults to localhost:11434 but can be configured for remote Ollama servers.
6
+ */
7
+ import { GatewayRequest, GatewayResponse, ProviderConfig, StreamChunk } from "../types.js";
8
+ import { BaseProvider } from "./base.js";
9
+ export declare class OllamaProvider extends BaseProvider {
10
+ constructor(config?: Partial<ProviderConfig>);
11
+ get id(): string;
12
+ get name(): string;
13
+ chat(request: GatewayRequest): Promise<GatewayResponse>;
14
+ chatStream(request: GatewayRequest): AsyncGenerator<StreamChunk>;
15
+ /**
16
+ * Check if Ollama server is reachable and get available models.
17
+ */
18
+ ping(): Promise<boolean>;
19
+ /**
20
+ * List models available in the local Ollama instance.
21
+ */
22
+ listLocalModels(): Promise<string[]>;
23
+ }
@@ -0,0 +1,187 @@
1
+ import { BaseProvider } from "./base.js";
2
+ export class OllamaProvider extends BaseProvider {
3
+ constructor(config) {
4
+ super({
5
+ id: "ollama",
6
+ name: "Ollama (Local LLM)",
7
+ baseUrl: process.env.OLLAMA_BASE_URL || "http://localhost:11434",
8
+ apiKey: "",
9
+ models: [
10
+ "llama3.3", "llama3.2", "llama3.1", "llama3",
11
+ "mistral", "mixtral", "codellama", "deepseek-coder",
12
+ "qwen2.5", "phi-4", "nemotron", "gemma2",
13
+ ],
14
+ defaultModel: process.env.OLLAMA_DEFAULT_MODEL || "llama3.3",
15
+ maxRetries: 2,
16
+ timeoutMs: 120000,
17
+ rpmLimit: 9999,
18
+ tpmLimit: 999999,
19
+ costPer1kInputTokens: 0,
20
+ costPer1kOutputTokens: 0,
21
+ enabled: true,
22
+ priority: 5,
23
+ weight: 1,
24
+ ...config,
25
+ });
26
+ }
27
+ get id() { return "ollama"; }
28
+ get name() { return "Ollama (Local LLM)"; }
29
+ async chat(request) {
30
+ const startTime = Date.now();
31
+ const model = request.model || this.config.defaultModel;
32
+ const traceId = request.traceId;
33
+ try {
34
+ const controller = new AbortController();
35
+ const timeout = setTimeout(() => controller.abort(), this.config.timeoutMs);
36
+ const response = await fetch(`${this.config.baseUrl}/api/chat`, {
37
+ method: "POST",
38
+ headers: { "Content-Type": "application/json" },
39
+ body: JSON.stringify({
40
+ model,
41
+ messages: [
42
+ ...(request.system ? [{ role: "system", content: request.system }] : []),
43
+ ...request.messages.map(m => ({
44
+ role: m.role,
45
+ content: m.content,
46
+ })),
47
+ ],
48
+ stream: false,
49
+ options: {
50
+ temperature: request.temperature ?? 0.7,
51
+ num_predict: request.maxTokens ?? 4096,
52
+ },
53
+ }),
54
+ signal: controller.signal,
55
+ });
56
+ clearTimeout(timeout);
57
+ if (!response.ok) {
58
+ const errorText = await response.text();
59
+ return this.errorResponse(`Ollama API error (${response.status}): ${errorText}`, model, traceId);
60
+ }
61
+ const data = await response.json();
62
+ const latencyMs = Date.now() - startTime;
63
+ const content = data.message?.content || "";
64
+ const inputTokens = data.prompt_eval_count || this.estimateTokens(request.messages.map(m => m.content).join(" "));
65
+ const outputTokens = data.eval_count || this.estimateTokens(content);
66
+ return {
67
+ success: true,
68
+ content,
69
+ model: data.model || model,
70
+ provider: "ollama",
71
+ usage: { inputTokens, outputTokens, totalTokens: inputTokens + outputTokens },
72
+ latencyMs,
73
+ cost: 0, // Local LLM is free
74
+ cached: false,
75
+ traceId,
76
+ };
77
+ }
78
+ catch (error) {
79
+ if (error instanceof Error && error.name === "AbortError") {
80
+ return this.timeoutError(model, this.config.timeoutMs, traceId);
81
+ }
82
+ return this.errorResponse(`Ollama request failed: ${error.message}. Is Ollama running on ${this.config.baseUrl}?`, model, traceId);
83
+ }
84
+ }
85
+ async *chatStream(request) {
86
+ const model = request.model || this.config.defaultModel;
87
+ try {
88
+ const response = await fetch(`${this.config.baseUrl}/api/chat`, {
89
+ method: "POST",
90
+ headers: { "Content-Type": "application/json" },
91
+ body: JSON.stringify({
92
+ model,
93
+ messages: [
94
+ ...(request.system ? [{ role: "system", content: request.system }] : []),
95
+ ...request.messages.map(m => ({
96
+ role: m.role,
97
+ content: m.content,
98
+ })),
99
+ ],
100
+ stream: true,
101
+ options: {
102
+ temperature: request.temperature ?? 0.7,
103
+ num_predict: request.maxTokens ?? 4096,
104
+ },
105
+ }),
106
+ });
107
+ if (!response.ok) {
108
+ yield { content: `Ollama stream error: ${response.status}`, done: true, provider: "ollama", model };
109
+ return;
110
+ }
111
+ const reader = response.body?.getReader();
112
+ if (!reader) {
113
+ yield { content: "No response body", done: true, provider: "ollama", model };
114
+ return;
115
+ }
116
+ const decoder = new TextDecoder();
117
+ let buffer = "";
118
+ while (true) {
119
+ const { done, value } = await reader.read();
120
+ if (done)
121
+ break;
122
+ buffer += decoder.decode(value, { stream: true });
123
+ const lines = buffer.split("\n");
124
+ buffer = lines.pop() || "";
125
+ for (const line of lines) {
126
+ if (!line.trim())
127
+ continue;
128
+ try {
129
+ const parsed = JSON.parse(line);
130
+ const content = parsed.message?.content || "";
131
+ if (content) {
132
+ yield { content, done: false, provider: "ollama", model };
133
+ }
134
+ if (parsed.done) {
135
+ yield { content: "", done: true, provider: "ollama", model };
136
+ return;
137
+ }
138
+ }
139
+ catch {
140
+ // Skip malformed
141
+ }
142
+ }
143
+ }
144
+ yield { content: "", done: true, provider: "ollama", model };
145
+ }
146
+ catch (error) {
147
+ yield {
148
+ content: `Ollama stream failed: ${error.message}`,
149
+ done: true,
150
+ provider: "ollama",
151
+ model,
152
+ };
153
+ }
154
+ }
155
+ /**
156
+ * Check if Ollama server is reachable and get available models.
157
+ */
158
+ async ping() {
159
+ try {
160
+ const response = await fetch(`${this.config.baseUrl}/api/tags`, {
161
+ signal: AbortSignal.timeout(5000),
162
+ });
163
+ return response.ok;
164
+ }
165
+ catch {
166
+ return false;
167
+ }
168
+ }
169
+ /**
170
+ * List models available in the local Ollama instance.
171
+ */
172
+ async listLocalModels() {
173
+ try {
174
+ const response = await fetch(`${this.config.baseUrl}/api/tags`, {
175
+ signal: AbortSignal.timeout(5000),
176
+ });
177
+ if (!response.ok)
178
+ return [];
179
+ const data = await response.json();
180
+ return data.models?.map(m => m.name) || [];
181
+ }
182
+ catch {
183
+ return [];
184
+ }
185
+ }
186
+ }
187
+ //# sourceMappingURL=ollama.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../../../../src/modules/gateway/providers/ollama.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,MAAM,OAAO,cAAe,SAAQ,YAAY;IAC5C,YAAY,MAAgC;QACxC,KAAK,CAAC;YACF,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,oBAAoB;YAC1B,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,wBAAwB;YAChE,MAAM,EAAE,EAAE;YACV,MAAM,EAAE;gBACJ,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ;gBAC5C,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,gBAAgB;gBACnD,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ;aAC3C;YACD,YAAY,EAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,UAAU;YAC5D,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,MAAM;YACjB,QAAQ,EAAE,IAAI;YACd,QAAQ,EAAE,MAAM;YAChB,oBAAoB,EAAE,CAAC;YACvB,qBAAqB,EAAE,CAAC;YACxB,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,CAAC;YACX,MAAM,EAAE,CAAC;YACT,GAAG,MAAM;SACZ,CAAC,CAAC;IACP,CAAC;IAED,IAAI,EAAE,KAAa,OAAO,QAAQ,CAAC,CAAC,CAAC;IACrC,IAAI,IAAI,KAAa,OAAO,oBAAoB,CAAC,CAAC,CAAC;IAEnD,KAAK,CAAC,IAAI,CAAC,OAAuB;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QACxD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAEhC,IAAI,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAE5E,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,EAAE;gBAC5D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK;oBACL,QAAQ,EAAE;wBACN,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBACjF,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BAC1B,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,OAAO,EAAE,CAAC,CAAC,OAAO;yBACrB,CAAC,CAAC;qBACN;oBACD,MAAM,EAAE,KAAK;oBACb,OAAO,EAAE;wBACL,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;wBACvC,WAAW,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;qBACzC;iBACJ,CAAC;gBACF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC5B,CAAC,CAAC;YAEH,YAAY,CAAC,OAAO,CAAC,CAAC;YAEtB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,aAAa,CACrB,qBAAqB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,EACrD,KAAK,EACL,OAAO,CACV,CAAC;YACN,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAKrC,CAAC;YAEI,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACzC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;YAC5C,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAClH,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAErE,OAAO;gBACH,OAAO,EAAE,IAAI;gBACb,OAAO;gBACP,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,KAAK;gBAC1B,QAAQ,EAAE,QAAQ;gBAClB,KAAK,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,GAAG,YAAY,EAAE;gBAC7E,SAAS;gBACT,IAAI,EAAE,CAAC,EAAE,oBAAoB;gBAC7B,MAAM,EAAE,KAAK;gBACb,OAAO;aACV,CAAC;QACN,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACxD,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACpE,CAAC;YACD,OAAO,IAAI,CAAC,aAAa,CACrB,0BAA2B,KAAe,CAAC,OAAO,0BAA0B,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,EAClG,KAAK,EACL,OAAO,CACV,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,OAAuB;QACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QAExD,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,EAAE;gBAC5D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK;oBACL,QAAQ,EAAE;wBACN,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBACjF,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BAC1B,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,OAAO,EAAE,CAAC,CAAC,OAAO;yBACrB,CAAC,CAAC;qBACN;oBACD,MAAM,EAAE,IAAI;oBACZ,OAAO,EAAE;wBACL,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;wBACvC,WAAW,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;qBACzC;iBACJ,CAAC;aACL,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,EAAE,OAAO,EAAE,wBAAwB,QAAQ,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;gBACpG,OAAO;YACX,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACV,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;gBAC7E,OAAO;YACX,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,OAAO,IAAI,EAAE,CAAC;gBACV,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACvB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;wBAAE,SAAS;oBAC3B,IAAI,CAAC;wBACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAGzC,CAAC;wBACU,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;wBAC9C,IAAI,OAAO,EAAE,CAAC;4BACV,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;wBAC9D,CAAC;wBACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;4BACd,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;4BAC7D,OAAO;wBACX,CAAC;oBACL,CAAC;oBAAC,MAAM,CAAC;wBACL,iBAAiB;oBACrB,CAAC;gBACL,CAAC;YACL,CAAC;YAED,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACjE,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,MAAM;gBACF,OAAO,EAAE,yBAA0B,KAAe,CAAC,OAAO,EAAE;gBAC5D,IAAI,EAAE,IAAI;gBACV,QAAQ,EAAE,QAAQ;gBAClB,KAAK;aACR,CAAC;QACN,CAAC;IACL,CAAC;IAED;;KAEC;IACD,KAAK,CAAC,IAAI;QACN,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,EAAE;gBAC5D,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;aACpC,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,EAAE,CAAC;QACvB,CAAC;QAAC,MAAM,CAAC;YACL,OAAO,KAAK,CAAC;QACjB,CAAC;IACL,CAAC;IAED;;KAEC;IACD,KAAK,CAAC,eAAe;QACjB,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,EAAE;gBAC5D,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;aACpC,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,EAAE;gBAAE,OAAO,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAErC,CAAC;YACI,OAAO,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACL,OAAO,EAAE,CAAC;QACd,CAAC;IACL,CAAC;CACJ"}
@@ -0,0 +1,14 @@
1
+ /**
2
+ * ATABEY LLM Gateway — OpenAI Provider
3
+ *
4
+ * Supports GPT-4, GPT-4o, GPT-4-turbo, GPT-3.5-turbo models via OpenAI API.
5
+ */
6
+ import { GatewayRequest, GatewayResponse, ProviderConfig, StreamChunk } from "../types.js";
7
+ import { BaseProvider } from "./base.js";
8
+ export declare class OpenAIProvider extends BaseProvider {
9
+ constructor(config?: Partial<ProviderConfig>);
10
+ get id(): string;
11
+ get name(): string;
12
+ chat(request: GatewayRequest): Promise<GatewayResponse>;
13
+ chatStream(request: GatewayRequest): AsyncGenerator<StreamChunk>;
14
+ }
@@ -0,0 +1,170 @@
1
+ import { BaseProvider } from "./base.js";
2
+ export class OpenAIProvider extends BaseProvider {
3
+ constructor(config) {
4
+ super({
5
+ id: "openai",
6
+ name: "OpenAI",
7
+ baseUrl: "https://api.openai.com/v1",
8
+ apiKey: process.env.OPENAI_API_KEY || "",
9
+ models: [
10
+ "gpt-4o", "gpt-4o-mini", "gpt-4-turbo", "gpt-4",
11
+ "gpt-3.5-turbo", "gpt-3.5-turbo-16k",
12
+ ],
13
+ defaultModel: "gpt-4o",
14
+ maxRetries: 3,
15
+ timeoutMs: 30000,
16
+ rpmLimit: 500,
17
+ tpmLimit: 200000,
18
+ costPer1kInputTokens: 0.01,
19
+ costPer1kOutputTokens: 0.03,
20
+ enabled: true,
21
+ priority: 1,
22
+ weight: 5,
23
+ ...config,
24
+ });
25
+ }
26
+ get id() { return "openai"; }
27
+ get name() { return "OpenAI"; }
28
+ async chat(request) {
29
+ const startTime = Date.now();
30
+ const model = request.model || this.config.defaultModel;
31
+ const traceId = request.traceId;
32
+ if (!this.config.apiKey) {
33
+ return this.errorResponse("OpenAI API key not configured. Set OPENAI_API_KEY env var.", model, traceId);
34
+ }
35
+ try {
36
+ const controller = new AbortController();
37
+ const timeout = setTimeout(() => controller.abort(), this.config.timeoutMs);
38
+ const response = await fetch(`${this.config.baseUrl}/chat/completions`, {
39
+ method: "POST",
40
+ headers: {
41
+ "Content-Type": "application/json",
42
+ Authorization: `Bearer ${this.config.apiKey}`,
43
+ },
44
+ body: JSON.stringify({
45
+ model,
46
+ messages: [
47
+ ...(request.system ? [{ role: "system", content: request.system }] : []),
48
+ ...request.messages.map(m => ({
49
+ role: m.role,
50
+ content: m.content,
51
+ })),
52
+ ],
53
+ temperature: request.temperature ?? 0.7,
54
+ max_tokens: request.maxTokens ?? 4096,
55
+ stream: false,
56
+ }),
57
+ signal: controller.signal,
58
+ });
59
+ clearTimeout(timeout);
60
+ if (!response.ok) {
61
+ const errorText = await response.text();
62
+ return this.errorResponse(`OpenAI API error (${response.status}): ${errorText}`, model, traceId);
63
+ }
64
+ const data = await response.json();
65
+ const latencyMs = Date.now() - startTime;
66
+ const inputTokens = data.usage?.prompt_tokens || this.estimateTokens(request.messages.map(m => m.content).join(" "));
67
+ const outputTokens = data.usage?.completion_tokens || this.estimateTokens(data.choices[0]?.message?.content || "");
68
+ const cost = (inputTokens / 1000) * this.config.costPer1kInputTokens +
69
+ (outputTokens / 1000) * this.config.costPer1kOutputTokens;
70
+ return {
71
+ success: true,
72
+ content: data.choices[0]?.message?.content || "",
73
+ model: data.model || model,
74
+ provider: "openai",
75
+ usage: {
76
+ inputTokens,
77
+ outputTokens,
78
+ totalTokens: inputTokens + outputTokens,
79
+ },
80
+ latencyMs,
81
+ cost,
82
+ cached: false,
83
+ traceId,
84
+ };
85
+ }
86
+ catch (error) {
87
+ if (error instanceof Error && error.name === "AbortError") {
88
+ return this.timeoutError(model, this.config.timeoutMs, traceId);
89
+ }
90
+ return this.errorResponse(`OpenAI request failed: ${error.message}`, model, traceId);
91
+ }
92
+ }
93
+ async *chatStream(request) {
94
+ const model = request.model || this.config.defaultModel;
95
+ if (!this.config.apiKey) {
96
+ yield { content: "OpenAI API key not configured.", done: true, provider: "openai", model };
97
+ return;
98
+ }
99
+ try {
100
+ const response = await fetch(`${this.config.baseUrl}/chat/completions`, {
101
+ method: "POST",
102
+ headers: {
103
+ "Content-Type": "application/json",
104
+ Authorization: `Bearer ${this.config.apiKey}`,
105
+ },
106
+ body: JSON.stringify({
107
+ model,
108
+ messages: [
109
+ ...(request.system ? [{ role: "system", content: request.system }] : []),
110
+ ...request.messages.map(m => ({
111
+ role: m.role,
112
+ content: m.content,
113
+ })),
114
+ ],
115
+ temperature: request.temperature ?? 0.7,
116
+ max_tokens: request.maxTokens ?? 4096,
117
+ stream: true,
118
+ }),
119
+ });
120
+ if (!response.ok) {
121
+ yield { content: `OpenAI stream error: ${response.status}`, done: true, provider: "openai", model };
122
+ return;
123
+ }
124
+ const reader = response.body?.getReader();
125
+ if (!reader) {
126
+ yield { content: "No response body", done: true, provider: "openai", model };
127
+ return;
128
+ }
129
+ const decoder = new TextDecoder();
130
+ let buffer = "";
131
+ while (true) {
132
+ const { done, value } = await reader.read();
133
+ if (done)
134
+ break;
135
+ buffer += decoder.decode(value, { stream: true });
136
+ const lines = buffer.split("\n");
137
+ buffer = lines.pop() || "";
138
+ for (const line of lines) {
139
+ if (line.startsWith("data: ")) {
140
+ const data = line.slice(6).trim();
141
+ if (data === "[DONE]") {
142
+ yield { content: "", done: true, provider: "openai", model };
143
+ return;
144
+ }
145
+ try {
146
+ const parsed = JSON.parse(data);
147
+ const content = parsed.choices?.[0]?.delta?.content || "";
148
+ if (content) {
149
+ yield { content, done: false, provider: "openai", model };
150
+ }
151
+ }
152
+ catch {
153
+ // Skip malformed JSON lines
154
+ }
155
+ }
156
+ }
157
+ }
158
+ yield { content: "", done: true, provider: "openai", model };
159
+ }
160
+ catch (error) {
161
+ yield {
162
+ content: `OpenAI stream failed: ${error.message}`,
163
+ done: true,
164
+ provider: "openai",
165
+ model,
166
+ };
167
+ }
168
+ }
169
+ }
170
+ //# sourceMappingURL=openai.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../../../src/modules/gateway/providers/openai.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,MAAM,OAAO,cAAe,SAAQ,YAAY;IAC5C,YAAY,MAAgC;QACxC,KAAK,CAAC;YACF,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,2BAA2B;YACpC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE;YACxC,MAAM,EAAE;gBACJ,QAAQ,EAAE,aAAa,EAAE,aAAa,EAAE,OAAO;gBAC/C,eAAe,EAAE,mBAAmB;aACvC;YACD,YAAY,EAAE,QAAQ;YACtB,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,GAAG;YACb,QAAQ,EAAE,MAAM;YAChB,oBAAoB,EAAE,IAAI;YAC1B,qBAAqB,EAAE,IAAI;YAC3B,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,CAAC;YACX,MAAM,EAAE,CAAC;YACT,GAAG,MAAM;SACZ,CAAC,CAAC;IACP,CAAC;IAED,IAAI,EAAE,KAAa,OAAO,QAAQ,CAAC,CAAC,CAAC;IACrC,IAAI,IAAI,KAAa,OAAO,QAAQ,CAAC,CAAC,CAAC;IAEvC,KAAK,CAAC,IAAI,CAAC,OAAuB;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QACxD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAEhC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,aAAa,CAAC,4DAA4D,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC5G,CAAC;QAED,IAAI,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAE5E,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,mBAAmB,EAAE;gBACpE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;iBAChD;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK;oBACL,QAAQ,EAAE;wBACN,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBACjF,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BAC1B,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,OAAO,EAAE,CAAC,CAAC,OAAO;yBACrB,CAAC,CAAC;qBACN;oBACD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;oBACvC,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;oBACrC,MAAM,EAAE,KAAK;iBAChB,CAAC;gBACF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC5B,CAAC,CAAC;YAEH,YAAY,CAAC,OAAO,CAAC,CAAC;YAEtB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,aAAa,CACrB,qBAAqB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,EACrD,KAAK,EACL,OAAO,CACV,CAAC;YACN,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAIrC,CAAC;YAEI,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACzC,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YACrH,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,EAAE,iBAAiB,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC;YACnH,MAAM,IAAI,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB;gBAC7D,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC;YAEjE,OAAO;gBACH,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE;gBAChD,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,KAAK;gBAC1B,QAAQ,EAAE,QAAQ;gBAClB,KAAK,EAAE;oBACH,WAAW;oBACX,YAAY;oBACZ,WAAW,EAAE,WAAW,GAAG,YAAY;iBAC1C;gBACD,SAAS;gBACT,IAAI;gBACJ,MAAM,EAAE,KAAK;gBACb,OAAO;aACV,CAAC;QACN,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACxD,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACpE,CAAC;YACD,OAAO,IAAI,CAAC,aAAa,CACrB,0BAA2B,KAAe,CAAC,OAAO,EAAE,EACpD,KAAK,EACL,OAAO,CACV,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,OAAuB;QACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QAExD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,MAAM,EAAE,OAAO,EAAE,gCAAgC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;YAC3F,OAAO;QACX,CAAC;QAED,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,mBAAmB,EAAE;gBACpE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;iBAChD;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK;oBACL,QAAQ,EAAE;wBACN,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBACjF,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BAC1B,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,OAAO,EAAE,CAAC,CAAC,OAAO;yBACrB,CAAC,CAAC;qBACN;oBACD,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;oBACvC,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;oBACrC,MAAM,EAAE,IAAI;iBACf,CAAC;aACL,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,EAAE,OAAO,EAAE,wBAAwB,QAAQ,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;gBACpG,OAAO;YACX,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACV,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;gBAC7E,OAAO;YACX,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,OAAO,IAAI,EAAE,CAAC;gBACV,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACvB,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAClC,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;4BACpB,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;4BAC7D,OAAO;wBACX,CAAC;wBACD,IAAI,CAAC;4BACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAE3C,CAAC;4BACY,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;4BAC1D,IAAI,OAAO,EAAE,CAAC;gCACV,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;4BAC9D,CAAC;wBACL,CAAC;wBAAC,MAAM,CAAC;4BACL,4BAA4B;wBAChC,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;YAED,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACjE,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,MAAM;gBACF,OAAO,EAAE,yBAA0B,KAAe,CAAC,OAAO,EAAE;gBAC5D,IAAI,EAAE,IAAI;gBACV,QAAQ,EAAE,QAAQ;gBAClB,KAAK;aACR,CAAC;QACN,CAAC;IACL,CAAC;CACJ"}
@@ -0,0 +1,41 @@
1
+ import { ProviderId } from "./types.js";
2
+ export declare class RateLimiter {
3
+ private states;
4
+ private readonly windowMs;
5
+ /** Custom limits set via initialize(), falls back to defaults */
6
+ private limits;
7
+ /**
8
+ * Initialize rate limiter with provider limits.
9
+ */
10
+ initialize(provider: ProviderId, rpmLimit: number, tpmLimit: number): void;
11
+ /**
12
+ * Check if a request is allowed for the given provider and token count.
13
+ * Returns true if allowed, false if rate limited.
14
+ */
15
+ check(provider: ProviderId, estimatedTokens: number): boolean;
16
+ /**
17
+ * Record a request for the given provider.
18
+ */
19
+ record(provider: ProviderId, tokens: number): void;
20
+ /**
21
+ * Get current usage stats for a provider.
22
+ */
23
+ getStats(provider: ProviderId): {
24
+ rpm: number;
25
+ tpm: number;
26
+ isLimited: boolean;
27
+ resetInMs: number;
28
+ } | null;
29
+ /**
30
+ * Reset window if time has elapsed.
31
+ */
32
+ private resetIfNeeded;
33
+ /**
34
+ * Get RPM limit for a provider (custom or default).
35
+ */
36
+ private getRpmLimit;
37
+ /**
38
+ * Get TPM limit for a provider (custom or default).
39
+ */
40
+ private getTpmLimit;
41
+ }
@@ -0,0 +1,119 @@
1
+ /**
2
+ * ATABEY LLM Gateway — Rate Limiter
3
+ *
4
+ * Token bucket algorithm for per-provider RPM and TPM limits.
5
+ * Prevents API quota exhaustion and 429 errors.
6
+ */
7
+ import { logger } from "../../shared/logger.js";
8
+ export class RateLimiter {
9
+ states = new Map();
10
+ windowMs = 60_000; // 1 minute window
11
+ /** Custom limits set via initialize(), falls back to defaults */
12
+ limits = new Map();
13
+ /**
14
+ * Initialize rate limiter with provider limits.
15
+ */
16
+ initialize(provider, rpmLimit, tpmLimit) {
17
+ this.states.set(provider, {
18
+ provider,
19
+ currentRpm: 0,
20
+ currentTpm: 0,
21
+ lastResetTime: Date.now(),
22
+ isLimited: false,
23
+ });
24
+ this.limits.set(provider, { rpm: rpmLimit, tpm: tpmLimit });
25
+ logger.debug(`[RATE] ${provider}: RPM=${rpmLimit}, TPM=${tpmLimit}`);
26
+ }
27
+ /**
28
+ * Check if a request is allowed for the given provider and token count.
29
+ * Returns true if allowed, false if rate limited.
30
+ */
31
+ check(provider, estimatedTokens) {
32
+ const state = this.states.get(provider);
33
+ if (!state)
34
+ return true; // Not initialized, allow
35
+ this.resetIfNeeded(state);
36
+ // Check limits
37
+ if (state.currentRpm >= this.getRpmLimit(provider)) {
38
+ state.isLimited = true;
39
+ return false;
40
+ }
41
+ if (state.currentTpm + estimatedTokens > this.getTpmLimit(provider)) {
42
+ state.isLimited = true;
43
+ return false;
44
+ }
45
+ return true;
46
+ }
47
+ /**
48
+ * Record a request for the given provider.
49
+ */
50
+ record(provider, tokens) {
51
+ const state = this.states.get(provider);
52
+ if (!state)
53
+ return;
54
+ this.resetIfNeeded(state);
55
+ state.currentRpm++;
56
+ state.currentTpm += tokens;
57
+ state.isLimited = false;
58
+ }
59
+ /**
60
+ * Get current usage stats for a provider.
61
+ */
62
+ getStats(provider) {
63
+ const state = this.states.get(provider);
64
+ if (!state)
65
+ return null;
66
+ this.resetIfNeeded(state);
67
+ return {
68
+ rpm: state.currentRpm,
69
+ tpm: state.currentTpm,
70
+ isLimited: state.isLimited,
71
+ resetInMs: this.windowMs - (Date.now() - state.lastResetTime),
72
+ };
73
+ }
74
+ /**
75
+ * Reset window if time has elapsed.
76
+ */
77
+ resetIfNeeded(state) {
78
+ const elapsed = Date.now() - state.lastResetTime;
79
+ if (elapsed >= this.windowMs) {
80
+ state.currentRpm = 0;
81
+ state.currentTpm = 0;
82
+ state.lastResetTime = Date.now();
83
+ state.isLimited = false;
84
+ }
85
+ }
86
+ /**
87
+ * Get RPM limit for a provider (custom or default).
88
+ */
89
+ getRpmLimit(provider) {
90
+ const custom = this.limits.get(provider);
91
+ if (custom)
92
+ return custom.rpm;
93
+ const defaults = {
94
+ openai: 500,
95
+ anthropic: 200,
96
+ google: 360,
97
+ grok: 100,
98
+ ollama: 9999,
99
+ };
100
+ return defaults[provider] || 100;
101
+ }
102
+ /**
103
+ * Get TPM limit for a provider (custom or default).
104
+ */
105
+ getTpmLimit(provider) {
106
+ const custom = this.limits.get(provider);
107
+ if (custom)
108
+ return custom.tpm;
109
+ const defaults = {
110
+ openai: 200000,
111
+ anthropic: 100000,
112
+ google: 360000,
113
+ grok: 50000,
114
+ ollama: 999999,
115
+ };
116
+ return defaults[provider] || 50000;
117
+ }
118
+ }
119
+ //# sourceMappingURL=rate-limiter.js.map