kailash 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

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 (37) hide show
  1. kailash/nodes/__init__.py +2 -1
  2. kailash/nodes/ai/__init__.py +26 -0
  3. kailash/nodes/ai/ai_providers.py +1272 -0
  4. kailash/nodes/ai/embedding_generator.py +853 -0
  5. kailash/nodes/ai/llm_agent.py +1166 -0
  6. kailash/nodes/api/auth.py +3 -3
  7. kailash/nodes/api/graphql.py +2 -2
  8. kailash/nodes/api/http.py +391 -44
  9. kailash/nodes/api/rate_limiting.py +2 -2
  10. kailash/nodes/api/rest.py +464 -56
  11. kailash/nodes/base.py +71 -12
  12. kailash/nodes/code/python.py +2 -1
  13. kailash/nodes/data/__init__.py +7 -0
  14. kailash/nodes/data/readers.py +28 -26
  15. kailash/nodes/data/retrieval.py +178 -0
  16. kailash/nodes/data/sharepoint_graph.py +7 -7
  17. kailash/nodes/data/sources.py +65 -0
  18. kailash/nodes/data/sql.py +4 -2
  19. kailash/nodes/data/writers.py +6 -3
  20. kailash/nodes/logic/operations.py +2 -1
  21. kailash/nodes/mcp/__init__.py +11 -0
  22. kailash/nodes/mcp/client.py +558 -0
  23. kailash/nodes/mcp/resource.py +682 -0
  24. kailash/nodes/mcp/server.py +571 -0
  25. kailash/nodes/transform/__init__.py +16 -1
  26. kailash/nodes/transform/chunkers.py +78 -0
  27. kailash/nodes/transform/formatters.py +96 -0
  28. kailash/runtime/docker.py +6 -6
  29. kailash/sdk_exceptions.py +24 -10
  30. kailash/tracking/metrics_collector.py +2 -1
  31. kailash/utils/templates.py +6 -6
  32. {kailash-0.1.1.dist-info → kailash-0.1.2.dist-info}/METADATA +344 -46
  33. {kailash-0.1.1.dist-info → kailash-0.1.2.dist-info}/RECORD +37 -26
  34. {kailash-0.1.1.dist-info → kailash-0.1.2.dist-info}/WHEEL +0 -0
  35. {kailash-0.1.1.dist-info → kailash-0.1.2.dist-info}/entry_points.txt +0 -0
  36. {kailash-0.1.1.dist-info → kailash-0.1.2.dist-info}/licenses/LICENSE +0 -0
  37. {kailash-0.1.1.dist-info → kailash-0.1.2.dist-info}/top_level.txt +0 -0
kailash/nodes/__init__.py CHANGED
@@ -1,7 +1,7 @@
1
1
  """Node system for the Kailash SDK."""
2
2
 
3
3
  # Import all node modules to ensure registration
4
- from kailash.nodes import ai, api, code, data, logic, transform
4
+ from kailash.nodes import ai, api, code, data, logic, mcp, transform
5
5
  from kailash.nodes.base import Node, NodeParameter, NodeRegistry, register_node
6
6
  from kailash.nodes.base_async import AsyncNode
7
7
  from kailash.nodes.code import PythonCodeNode
@@ -19,5 +19,6 @@ __all__ = [
19
19
  "code",
20
20
  "data",
21
21
  "logic",
22
+ "mcp",
22
23
  "transform",
23
24
  ]
kailash/nodes/ai/__init__.py CHANGED
@@ -1,6 +1,20 @@
1
1
  """AI and ML nodes for the Kailash SDK."""
2
2
 
3
3
  from .agents import ChatAgent, FunctionCallingAgent, PlanningAgent, RetrievalAgent
4
+
5
+ # Import from unified ai_providers module
6
+ from .ai_providers import (
7
+ PROVIDERS,
8
+ AnthropicProvider,
9
+ LLMProvider,
10
+ MockProvider,
11
+ OllamaProvider,
12
+ OpenAIProvider,
13
+ get_available_providers,
14
+ get_provider,
15
+ )
16
+ from .embedding_generator import EmbeddingGenerator
17
+ from .llm_agent import LLMAgent
4
18
  from .models import (
5
19
  ModelPredictor,
6
20
  NamedEntityRecognizer,
@@ -16,6 +30,18 @@ __all__ = [
16
30
  "RetrievalAgent",
17
31
  "FunctionCallingAgent",
18
32
  "PlanningAgent",
33
+ "LLMAgent",
34
+ # Embedding and Vector Operations
35
+ "EmbeddingGenerator",
36
+ # Provider Infrastructure
37
+ "LLMProvider",
38
+ "OllamaProvider",
39
+ "OpenAIProvider",
40
+ "AnthropicProvider",
41
+ "MockProvider",
42
+ "get_provider",
43
+ "get_available_providers",
44
+ "PROVIDERS",
19
45
  # Models
20
46
  "TextClassifier",
21
47
  "TextEmbedder",