mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 00:16:23 +02:00
a9197d11ee
- Keeps processing in different flows separate so that data can go to different stores / collections etc. - Potentially supports different processing flows - Tidies the processing API with common base-classes for e.g. LLMs, and automatic configuration of 'clients' to use the right queue names in a flow
93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
|
|
import json
|
|
|
|
from . request_response_spec import RequestResponse, RequestResponseSpec
|
|
from .. schema import PromptRequest, PromptResponse
|
|
|
|
class PromptClient(RequestResponse):
|
|
|
|
async def prompt(self, id, variables, timeout=600):
|
|
|
|
resp = await self.request(
|
|
PromptRequest(
|
|
id = id,
|
|
terms = {
|
|
k: json.dumps(v)
|
|
for k, v in variables.items()
|
|
}
|
|
),
|
|
timeout=timeout
|
|
)
|
|
|
|
if resp.error:
|
|
raise RuntimeError(resp.error.message)
|
|
|
|
if resp.text: return resp.text
|
|
|
|
return json.loads(resp.object)
|
|
|
|
async def extract_definitions(self, text, timeout=600):
|
|
return await self.prompt(
|
|
id = "extract-definitions",
|
|
variables = { "text": text },
|
|
timeout = timeout,
|
|
)
|
|
|
|
async def extract_relationships(self, text, timeout=600):
|
|
return await self.prompt(
|
|
id = "extract-relationships",
|
|
variables = { "text": text },
|
|
timeout = timeout,
|
|
)
|
|
|
|
async def kg_prompt(self, query, kg, timeout=600):
|
|
return await self.prompt(
|
|
id = "kg-prompt",
|
|
variables = {
|
|
"query": query,
|
|
"knowledge": [
|
|
{ "s": v[0], "p": v[1], "o": v[2] }
|
|
for v in kg
|
|
]
|
|
},
|
|
timeout = timeout,
|
|
)
|
|
|
|
async def document_prompt(self, query, documents, timeout=600):
|
|
return await self.prompt(
|
|
id = "document-prompt",
|
|
variables = {
|
|
"query": query,
|
|
"documents": documents,
|
|
},
|
|
timeout = timeout,
|
|
)
|
|
|
|
async def agent_react(self, variables, timeout=600):
|
|
return await self.prompt(
|
|
id = "agent-react",
|
|
variables = variables,
|
|
timeout = timeout,
|
|
)
|
|
|
|
async def question(self, question, timeout=600):
|
|
return await self.prompt(
|
|
id = "question",
|
|
variables = {
|
|
"question": question,
|
|
},
|
|
timeout = timeout,
|
|
)
|
|
|
|
class PromptClientSpec(RequestResponseSpec):
|
|
def __init__(
|
|
self, request_name, response_name,
|
|
):
|
|
super(PromptClientSpec, self).__init__(
|
|
request_name = request_name,
|
|
request_schema = PromptRequest,
|
|
response_name = response_name,
|
|
response_schema = PromptResponse,
|
|
impl = PromptClient,
|
|
)
|
|
|