From e2d7778cd349b0d828ceb690c8834a51c4cd60a3 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Thu, 17 Jul 2025 20:45:36 +0100 Subject: [PATCH] Fix broken imports --- trustgraph-flow/setup.py | 2 -- .../trustgraph/prompt/template/service.py | 14 +++++++------- .../trustgraph/template/prompt_manager.py | 2 +- 3 files changed, 8 insertions(+), 10 deletions(-) diff --git a/trustgraph-flow/setup.py b/trustgraph-flow/setup.py index b91c69dd..59b94adc 100644 --- a/trustgraph-flow/setup.py +++ b/trustgraph-flow/setup.py @@ -97,7 +97,6 @@ setuptools.setup( "scripts/kg-extract-definitions", "scripts/kg-extract-relationships", "scripts/kg-extract-agent", -# "scripts/kg-extract-topics", "scripts/kg-store", "scripts/kg-manager", "scripts/librarian", @@ -107,7 +106,6 @@ setuptools.setup( "scripts/oe-write-milvus", "scripts/pdf-decoder", "scripts/pdf-ocr-mistral", - "scripts/prompt-generic", "scripts/prompt-template", "scripts/rows-write-cassandra", "scripts/run-processing", diff --git a/trustgraph-flow/trustgraph/prompt/template/service.py b/trustgraph-flow/trustgraph/prompt/template/service.py index 02395d6d..4033ed8a 100755 --- a/trustgraph-flow/trustgraph/prompt/template/service.py +++ b/trustgraph-flow/trustgraph/prompt/template/service.py @@ -7,15 +7,15 @@ import asyncio import json import re -from ....schema import Definition, Relationship, Triple -from ....schema import Topic -from ....schema import PromptRequest, PromptResponse, Error -from ....schema import TextCompletionRequest, TextCompletionResponse +from ...schema import Definition, Relationship, Triple +from ...schema import Topic +from ...schema import PromptRequest, PromptResponse, Error +from ...schema import TextCompletionRequest, TextCompletionResponse -from ....base import FlowProcessor -from ....base import ProducerSpec, ConsumerSpec, TextCompletionClientSpec +from ...base import FlowProcessor +from ...base import ProducerSpec, ConsumerSpec, TextCompletionClientSpec -from .template import PromptConfiguration, Prompt, PromptManager +from ...template import PromptConfiguration, Prompt, PromptManager default_ident = "prompt" default_concurrency = 1 diff --git a/trustgraph-flow/trustgraph/template/prompt_manager.py b/trustgraph-flow/trustgraph/template/prompt_manager.py index 844f3b0b..25830f55 100644 --- a/trustgraph-flow/trustgraph/template/prompt_manager.py +++ b/trustgraph-flow/trustgraph/template/prompt_manager.py @@ -19,7 +19,7 @@ class Prompt: class PromptManager: - def __init__(self, {}): + def __init__(self): self.config = PromptConfiguration("", {}, {}) self.terms = config.global_terms