omni_agent 0.1.11 → 0.2.0

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 (75) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +43 -0
  3. data/README.md +96 -2
  4. data/Rakefile +4 -0
  5. data/app/jobs/omni_agent/embed_job.rb +10 -0
  6. data/app/jobs/omni_agent/ingest_job.rb +12 -0
  7. data/docs/agent/callbacks.md +60 -0
  8. data/docs/agent/delegation.md +56 -0
  9. data/docs/agent/provider.md +93 -0
  10. data/docs/agent/run-entrypoints.md +81 -0
  11. data/docs/agent/streaming.md +62 -0
  12. data/docs/agent/structured-output.md +80 -0
  13. data/docs/configuration.md +68 -0
  14. data/docs/evals/caching.md +21 -0
  15. data/docs/evals/generating-and-running.md +87 -0
  16. data/docs/evals/golden-sets.md +32 -0
  17. data/docs/evals/judge-provider.md +20 -0
  18. data/docs/generators.md +143 -0
  19. data/docs/index.md +98 -0
  20. data/docs/intro.md +93 -0
  21. data/docs/mcp/authentication.md +226 -0
  22. data/docs/mcp/defining-servers.md +196 -0
  23. data/docs/mcp/transports.md +166 -0
  24. data/docs/rag.md +372 -0
  25. data/docs/stock-agents/ocr.md +52 -0
  26. data/docs/stock-agents/overview.md +91 -0
  27. data/docs/supported-providers.md +47 -0
  28. data/docs/tools.md +171 -0
  29. data/exe/omni_agent +48 -10
  30. data/lib/generators/omni_agent/chunks/chunks_generator.rb +130 -0
  31. data/lib/generators/omni_agent/embeddings/embeddings_generator.rb +72 -0
  32. data/lib/generators/omni_agent/install/install_generator.rb +78 -1
  33. data/lib/generators/omni_agent/stock_agent/stock_agent_generator.rb +73 -0
  34. data/lib/generators/omni_agent/vector_store.rb +113 -0
  35. data/lib/omni_agent/agent.rb +168 -39
  36. data/lib/omni_agent/chunkable.rb +181 -0
  37. data/lib/omni_agent/chunkers/hierarchical.rb +20 -0
  38. data/lib/omni_agent/chunkers/markdown.rb +41 -0
  39. data/lib/omni_agent/chunkers/recursive.rb +53 -0
  40. data/lib/omni_agent/chunkers.rb +31 -0
  41. data/lib/omni_agent/configuration.rb +14 -1
  42. data/lib/omni_agent/docs.rb +32 -0
  43. data/lib/omni_agent/embeddable.rb +131 -0
  44. data/lib/omni_agent/engine.rb +11 -0
  45. data/lib/omni_agent/errors.rb +15 -0
  46. data/lib/omni_agent/loaders/html.rb +21 -0
  47. data/lib/omni_agent/loaders/markdown.rb +18 -0
  48. data/lib/omni_agent/loaders/pdf.rb +25 -0
  49. data/lib/omni_agent/loaders/text.rb +15 -0
  50. data/lib/omni_agent/loaders.rb +68 -0
  51. data/lib/omni_agent/providers/base.rb +14 -0
  52. data/lib/omni_agent/providers/mock.rb +89 -4
  53. data/lib/omni_agent/providers/ollama.rb +4 -0
  54. data/lib/omni_agent/providers/openai.rb +33 -2
  55. data/lib/omni_agent/providers/response.rb +6 -1
  56. data/lib/omni_agent/providers.rb +7 -0
  57. data/lib/omni_agent/schema.rb +200 -0
  58. data/lib/omni_agent/semantic_search.rb +45 -0
  59. data/lib/omni_agent/stock_agent.rb +59 -0
  60. data/lib/omni_agent/stock_agents/ocr/prompt.md.erb +6 -0
  61. data/lib/omni_agent/stock_agents/ocr.rb +70 -0
  62. data/lib/omni_agent/stock_agents.rb +26 -0
  63. data/lib/omni_agent/tool.rb +7 -107
  64. data/lib/omni_agent/version.rb +1 -1
  65. data/lib/omni_agent.rb +12 -1
  66. data/lib/tasks/omni_agent_chunks_tasks.rake +16 -0
  67. data/lib/tasks/omni_agent_embeddings_tasks.rake +15 -0
  68. metadata +49 -8
  69. data/app/assets/stylesheets/omni_agent/application.css +0 -15
  70. data/app/controllers/omni_agent/application_controller.rb +0 -4
  71. data/app/helpers/omni_agent/application_helper.rb +0 -4
  72. data/app/jobs/omni_agent/application_job.rb +0 -4
  73. data/app/mailers/omni_agent/application_mailer.rb +0 -6
  74. data/app/models/omni_agent/application_record.rb +0 -5
  75. data/app/views/layouts/omni_agent/application.html.erb +0 -17
@@ -0,0 +1,16 @@
1
+ namespace :omni_agent do
2
+ namespace :chunks do
3
+ desc 'Chunk and embed records that have no chunks. Usage: rake "omni_agent:chunks:ingest[Document]", add ",force" to re-ingest every record'
4
+ task :ingest, [ :model, :force ] => :environment do |_task, args|
5
+ model_name = args[:model].to_s.strip
6
+ abort 'Please provide a model. Example: bin/rails "omni_agent:chunks:ingest[Document]"' if model_name.empty?
7
+
8
+ model = model_name.safe_constantize
9
+ abort "#{model_name} does not use has_chunks" unless model.respond_to?(:ingest_all)
10
+
11
+ skipped = model.ingest_all(force: args[:force] == "force")
12
+ skipped.each { |id, message| puts "Skipped #{model_name}##{id}: #{message}" }
13
+ puts "Ingested #{model_name} records."
14
+ end
15
+ end
16
+ end
@@ -0,0 +1,15 @@
1
+ namespace :omni_agent do
2
+ namespace :embeddings do
3
+ desc 'Embed records missing a vector. Usage: rake "omni_agent:embeddings:backfill[Document]", add ",force" to re-embed every record'
4
+ task :backfill, [ :model, :force ] => :environment do |_task, args|
5
+ model_name = args[:model].to_s.strip
6
+ abort 'Please provide a model. Example: bin/rails "omni_agent:embeddings:backfill[Document]"' if model_name.empty?
7
+
8
+ model = model_name.safe_constantize
9
+ abort "#{model_name} does not use has_embedding" unless model.respond_to?(:embed_all)
10
+
11
+ model.embed_all(force: args[:force] == "force")
12
+ puts "Embedded #{model_name} records."
13
+ end
14
+ end
15
+ end
metadata CHANGED
@@ -1,7 +1,7 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: omni_agent
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.1.11
4
+ version: 0.2.0
5
5
  platform: ruby
6
6
  authors:
7
7
  - ACR1209
@@ -50,22 +50,50 @@ files:
50
50
  - MIT-LICENSE
51
51
  - README.md
52
52
  - Rakefile
53
- - app/assets/stylesheets/omni_agent/application.css
54
- - app/controllers/omni_agent/application_controller.rb
55
- - app/helpers/omni_agent/application_helper.rb
56
- - app/jobs/omni_agent/application_job.rb
57
- - app/mailers/omni_agent/application_mailer.rb
58
- - app/models/omni_agent/application_record.rb
59
- - app/views/layouts/omni_agent/application.html.erb
53
+ - app/jobs/omni_agent/embed_job.rb
54
+ - app/jobs/omni_agent/ingest_job.rb
60
55
  - config/routes.rb
56
+ - docs/agent/callbacks.md
57
+ - docs/agent/delegation.md
58
+ - docs/agent/provider.md
59
+ - docs/agent/run-entrypoints.md
60
+ - docs/agent/streaming.md
61
+ - docs/agent/structured-output.md
62
+ - docs/configuration.md
63
+ - docs/evals/caching.md
64
+ - docs/evals/generating-and-running.md
65
+ - docs/evals/golden-sets.md
66
+ - docs/evals/judge-provider.md
67
+ - docs/generators.md
68
+ - docs/index.md
69
+ - docs/intro.md
70
+ - docs/mcp/authentication.md
71
+ - docs/mcp/defining-servers.md
72
+ - docs/mcp/transports.md
73
+ - docs/rag.md
74
+ - docs/stock-agents/ocr.md
75
+ - docs/stock-agents/overview.md
76
+ - docs/supported-providers.md
77
+ - docs/tools.md
61
78
  - exe/omni_agent
62
79
  - lib/generators/omni_agent/agent/agent_generator.rb
80
+ - lib/generators/omni_agent/chunks/chunks_generator.rb
81
+ - lib/generators/omni_agent/embeddings/embeddings_generator.rb
63
82
  - lib/generators/omni_agent/eval/eval_generator.rb
64
83
  - lib/generators/omni_agent/install/install_generator.rb
65
84
  - lib/generators/omni_agent/mcp_server/mcp_server_generator.rb
85
+ - lib/generators/omni_agent/stock_agent/stock_agent_generator.rb
86
+ - lib/generators/omni_agent/vector_store.rb
66
87
  - lib/omni_agent.rb
67
88
  - lib/omni_agent/agent.rb
89
+ - lib/omni_agent/chunkable.rb
90
+ - lib/omni_agent/chunkers.rb
91
+ - lib/omni_agent/chunkers/hierarchical.rb
92
+ - lib/omni_agent/chunkers/markdown.rb
93
+ - lib/omni_agent/chunkers/recursive.rb
68
94
  - lib/omni_agent/configuration.rb
95
+ - lib/omni_agent/docs.rb
96
+ - lib/omni_agent/embeddable.rb
69
97
  - lib/omni_agent/engine.rb
70
98
  - lib/omni_agent/errors.rb
71
99
  - lib/omni_agent/eval.rb
@@ -82,6 +110,11 @@ files:
82
110
  - lib/omni_agent/eval/run.rb
83
111
  - lib/omni_agent/eval/runner.rb
84
112
  - lib/omni_agent/eval/tool_call_assertion.rb
113
+ - lib/omni_agent/loaders.rb
114
+ - lib/omni_agent/loaders/html.rb
115
+ - lib/omni_agent/loaders/markdown.rb
116
+ - lib/omni_agent/loaders/pdf.rb
117
+ - lib/omni_agent/loaders/text.rb
85
118
  - lib/omni_agent/mcp.rb
86
119
  - lib/omni_agent/mcp/authenticator.rb
87
120
  - lib/omni_agent/mcp/handler.rb
@@ -105,12 +138,20 @@ files:
105
138
  - lib/omni_agent/providers/openai.rb
106
139
  - lib/omni_agent/providers/response.rb
107
140
  - lib/omni_agent/railtie.rb
141
+ - lib/omni_agent/schema.rb
142
+ - lib/omni_agent/semantic_search.rb
143
+ - lib/omni_agent/stock_agent.rb
144
+ - lib/omni_agent/stock_agents.rb
145
+ - lib/omni_agent/stock_agents/ocr.rb
146
+ - lib/omni_agent/stock_agents/ocr/prompt.md.erb
108
147
  - lib/omni_agent/streaming/event.rb
109
148
  - lib/omni_agent/streaming/proxy.rb
110
149
  - lib/omni_agent/tasks/omni_agent_tasks.rake
111
150
  - lib/omni_agent/tool.rb
112
151
  - lib/omni_agent/tool/schema_builder.rb
113
152
  - lib/omni_agent/version.rb
153
+ - lib/tasks/omni_agent_chunks_tasks.rake
154
+ - lib/tasks/omni_agent_embeddings_tasks.rake
114
155
  - lib/tasks/omni_agent_eval_tasks.rake
115
156
  - lib/tasks/omni_agent_mcp_tasks.rake
116
157
  homepage: https://github.com/ACR1209/omni_agent
@@ -1,15 +0,0 @@
1
- /*
2
- * This is a manifest file that'll be compiled into application.css, which will include all the files
3
- * listed below.
4
- *
5
- * Any CSS and SCSS file within this directory, lib/assets/stylesheets, vendor/assets/stylesheets,
6
- * or any plugin's vendor/assets/stylesheets directory can be referenced here using a relative path.
7
- *
8
- * You're free to add application-wide styles to this file and they'll appear at the bottom of the
9
- * compiled file so the styles you add here take precedence over styles defined in any other CSS/SCSS
10
- * files in this directory. Styles in this file should be added after the last require_* statement.
11
- * It is generally better to create a new file per style scope.
12
- *
13
- *= require_tree .
14
- *= require_self
15
- */
@@ -1,4 +0,0 @@
1
- module OmniAgent
2
- class ApplicationController < ActionController::Base
3
- end
4
- end
@@ -1,4 +0,0 @@
1
- module OmniAgent
2
- module ApplicationHelper
3
- end
4
- end
@@ -1,4 +0,0 @@
1
- module OmniAgent
2
- class ApplicationJob < ActiveJob::Base
3
- end
4
- end
@@ -1,6 +0,0 @@
1
- module OmniAgent
2
- class ApplicationMailer < ActionMailer::Base
3
- default from: "from@example.com"
4
- layout "mailer"
5
- end
6
- end
@@ -1,5 +0,0 @@
1
- module OmniAgent
2
- class ApplicationRecord < ActiveRecord::Base
3
- self.abstract_class = true
4
- end
5
- end
@@ -1,17 +0,0 @@
1
- <!DOCTYPE html>
2
- <html>
3
- <head>
4
- <title>Omni agent</title>
5
- <%= csrf_meta_tags %>
6
- <%= csp_meta_tag %>
7
-
8
- <%= yield :head %>
9
-
10
- <%= stylesheet_link_tag "omni_agent/application", media: "all" %>
11
- </head>
12
- <body>
13
-
14
- <%= yield %>
15
-
16
- </body>
17
- </html>