ragspark 0.3.0__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.
- ragspark/__init__.py +75 -0
- ragspark/chunker/__init__.py +878 -0
- ragspark/cli.py +122 -0
- ragspark/embedder/__init__.py +335 -0
- ragspark/llm/__init__.py +812 -0
- ragspark/parser/__init__.py +147 -0
- ragspark/parser/cleaner.py +149 -0
- ragspark/parser/core.py +303 -0
- ragspark/parser/engines.py +410 -0
- ragspark/parser/legal_regex.py +635 -0
- ragspark/parser/models.py +266 -0
- ragspark/prompt_eng/__init__.py +897 -0
- ragspark/py.typed +2 -0
- ragspark/rag/__init__.py +814 -0
- ragspark/reranker/__init__.py +183 -0
- ragspark/ui/__init__.py +734 -0
- ragspark/vectordb/__init__.py +704 -0
- ragspark-0.3.0.dist-info/METADATA +258 -0
- ragspark-0.3.0.dist-info/RECORD +21 -0
- ragspark-0.3.0.dist-info/WHEEL +4 -0
- ragspark-0.3.0.dist-info/entry_points.txt +2 -0
ragspark/__init__.py
ADDED
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
"""
|
|
2
|
+
`ragspark` — Production Unified RAG Pipeline Library (General + Specialized Law RAG + Chatbot Studio UI).
|
|
3
|
+
|
|
4
|
+
Import any pipeline step directly:
|
|
5
|
+
from ragspark import parser
|
|
6
|
+
from ragspark import chunker
|
|
7
|
+
from ragspark import embedder
|
|
8
|
+
from ragspark import vectordb
|
|
9
|
+
from ragspark import prompt_eng
|
|
10
|
+
from ragspark import llm
|
|
11
|
+
from ragspark import rag
|
|
12
|
+
from ragspark import ui
|
|
13
|
+
"""
|
|
14
|
+
|
|
15
|
+
from __future__ import annotations
|
|
16
|
+
|
|
17
|
+
from ragspark import chunker, embedder, llm, parser, prompt_eng, rag, reranker, ui, vectordb
|
|
18
|
+
from ragspark.chunker import (
|
|
19
|
+
Chunk,
|
|
20
|
+
chunk,
|
|
21
|
+
chunk_combination,
|
|
22
|
+
chunk_law,
|
|
23
|
+
)
|
|
24
|
+
from ragspark.embedder import embed, embed_query
|
|
25
|
+
from ragspark.parser import ParsedDocument, parse, parse_law, parse_text, parse_unstructured
|
|
26
|
+
from ragspark.prompt_eng import BuiltPrompt, detect_language
|
|
27
|
+
from ragspark.rag import RAGPipeline, RAGResponse, ask, ask_law, chat
|
|
28
|
+
from ragspark.rag import create as build_rag
|
|
29
|
+
from ragspark.rag import create_law as build_law_rag
|
|
30
|
+
from ragspark.reranker import Reranker, rerank
|
|
31
|
+
from ragspark.ui import launch as launch_ui
|
|
32
|
+
from ragspark.ui import notebook_chat
|
|
33
|
+
from ragspark.vectordb import SearchResult, VectorStore, store
|
|
34
|
+
|
|
35
|
+
__version__ = "0.3.0"
|
|
36
|
+
|
|
37
|
+
__all__ = [
|
|
38
|
+
"__version__",
|
|
39
|
+
"parser",
|
|
40
|
+
"chunker",
|
|
41
|
+
"embedder",
|
|
42
|
+
"vectordb",
|
|
43
|
+
"prompt_eng",
|
|
44
|
+
"llm",
|
|
45
|
+
"rag",
|
|
46
|
+
"ui",
|
|
47
|
+
"reranker",
|
|
48
|
+
"rerank",
|
|
49
|
+
"Reranker",
|
|
50
|
+
"launch_ui",
|
|
51
|
+
"notebook_chat",
|
|
52
|
+
"store",
|
|
53
|
+
"parse",
|
|
54
|
+
"parse_law",
|
|
55
|
+
"parse_unstructured",
|
|
56
|
+
"parse_text",
|
|
57
|
+
"chunk",
|
|
58
|
+
"chunk_combination",
|
|
59
|
+
"chunk_law",
|
|
60
|
+
"embed",
|
|
61
|
+
"embed_query",
|
|
62
|
+
"build_rag",
|
|
63
|
+
"build_law_rag",
|
|
64
|
+
"ask",
|
|
65
|
+
"chat",
|
|
66
|
+
"ask_law",
|
|
67
|
+
"detect_language",
|
|
68
|
+
"ParsedDocument",
|
|
69
|
+
"Chunk",
|
|
70
|
+
"VectorStore",
|
|
71
|
+
"SearchResult",
|
|
72
|
+
"BuiltPrompt",
|
|
73
|
+
"RAGPipeline",
|
|
74
|
+
"RAGResponse",
|
|
75
|
+
]
|