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.
- checksums.yaml +4 -4
- data/CHANGELOG.md +43 -0
- data/README.md +96 -2
- data/Rakefile +4 -0
- data/app/jobs/omni_agent/embed_job.rb +10 -0
- data/app/jobs/omni_agent/ingest_job.rb +12 -0
- data/docs/agent/callbacks.md +60 -0
- data/docs/agent/delegation.md +56 -0
- data/docs/agent/provider.md +93 -0
- data/docs/agent/run-entrypoints.md +81 -0
- data/docs/agent/streaming.md +62 -0
- data/docs/agent/structured-output.md +80 -0
- data/docs/configuration.md +68 -0
- data/docs/evals/caching.md +21 -0
- data/docs/evals/generating-and-running.md +87 -0
- data/docs/evals/golden-sets.md +32 -0
- data/docs/evals/judge-provider.md +20 -0
- data/docs/generators.md +143 -0
- data/docs/index.md +98 -0
- data/docs/intro.md +93 -0
- data/docs/mcp/authentication.md +226 -0
- data/docs/mcp/defining-servers.md +196 -0
- data/docs/mcp/transports.md +166 -0
- data/docs/rag.md +372 -0
- data/docs/stock-agents/ocr.md +52 -0
- data/docs/stock-agents/overview.md +91 -0
- data/docs/supported-providers.md +47 -0
- data/docs/tools.md +171 -0
- data/exe/omni_agent +48 -10
- data/lib/generators/omni_agent/chunks/chunks_generator.rb +130 -0
- data/lib/generators/omni_agent/embeddings/embeddings_generator.rb +72 -0
- data/lib/generators/omni_agent/install/install_generator.rb +78 -1
- data/lib/generators/omni_agent/stock_agent/stock_agent_generator.rb +73 -0
- data/lib/generators/omni_agent/vector_store.rb +113 -0
- data/lib/omni_agent/agent.rb +168 -39
- data/lib/omni_agent/chunkable.rb +181 -0
- data/lib/omni_agent/chunkers/hierarchical.rb +20 -0
- data/lib/omni_agent/chunkers/markdown.rb +41 -0
- data/lib/omni_agent/chunkers/recursive.rb +53 -0
- data/lib/omni_agent/chunkers.rb +31 -0
- data/lib/omni_agent/configuration.rb +14 -1
- data/lib/omni_agent/docs.rb +32 -0
- data/lib/omni_agent/embeddable.rb +131 -0
- data/lib/omni_agent/engine.rb +11 -0
- data/lib/omni_agent/errors.rb +15 -0
- data/lib/omni_agent/loaders/html.rb +21 -0
- data/lib/omni_agent/loaders/markdown.rb +18 -0
- data/lib/omni_agent/loaders/pdf.rb +25 -0
- data/lib/omni_agent/loaders/text.rb +15 -0
- data/lib/omni_agent/loaders.rb +68 -0
- data/lib/omni_agent/providers/base.rb +14 -0
- data/lib/omni_agent/providers/mock.rb +89 -4
- data/lib/omni_agent/providers/ollama.rb +4 -0
- data/lib/omni_agent/providers/openai.rb +33 -2
- data/lib/omni_agent/providers/response.rb +6 -1
- data/lib/omni_agent/providers.rb +7 -0
- data/lib/omni_agent/schema.rb +200 -0
- data/lib/omni_agent/semantic_search.rb +45 -0
- data/lib/omni_agent/stock_agent.rb +59 -0
- data/lib/omni_agent/stock_agents/ocr/prompt.md.erb +6 -0
- data/lib/omni_agent/stock_agents/ocr.rb +70 -0
- data/lib/omni_agent/stock_agents.rb +26 -0
- data/lib/omni_agent/tool.rb +7 -107
- data/lib/omni_agent/version.rb +1 -1
- data/lib/omni_agent.rb +12 -1
- data/lib/tasks/omni_agent_chunks_tasks.rake +16 -0
- data/lib/tasks/omni_agent_embeddings_tasks.rake +15 -0
- metadata +49 -8
- data/app/assets/stylesheets/omni_agent/application.css +0 -15
- data/app/controllers/omni_agent/application_controller.rb +0 -4
- data/app/helpers/omni_agent/application_helper.rb +0 -4
- data/app/jobs/omni_agent/application_job.rb +0 -4
- data/app/mailers/omni_agent/application_mailer.rb +0 -6
- data/app/models/omni_agent/application_record.rb +0 -5
- 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.
|
|
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/
|
|
54
|
-
- app/
|
|
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,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>
|