Skip to content

Documenter

DocumenterAgent

Bases: Agent

An agent that generates and renders documents based on templates, by running natural language and SQL queries, specified as propertiesconstructed by substituting input, and other contextual information defined as variables.

Properties (in addition to Agent properties):

Name Type Default Description
hilite str "" (Optional) A template string for highlighting the document using the HILITER agent. If specified, the document will be highlighted before rendering.
template str or dict "" A template string or JSON object for the document. This template will be processed by substituting variables and results from queries.
questions dict {} (Optional) A dictionary of natural language questions to be processed by the NL2SQL agent. Each key is a question name, and the value is the question template string.
queries dict {} (Optional) A dictionary of SQL queries to be executed by the QUERYEXECUTOR agent. Each key is a query name, and the value is the SQL query template string.

Inputs: - DEFAULT: Accepts user input text to initiate document generation.

Outputs: - DEFAULT: Outputs the generated document, optionally highlighted, tagged as DOC.

Source code in blue/agents/documenter.py
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
class DocumenterAgent(Agent):
    """An agent that generates and renders documents based on templates, by running natural language and SQL queries, specified as propertiesconstructed by substituting input, and other contextual information defined as variables.

    Properties (in addition to Agent properties):
    ----------
    | Name           | Type                 | Default | Description |
    |----------------|--------------------|----------|---------|
    | `hilite`       | `str`            | `""`     | (Optional) A template string for highlighting the document using the HILITER agent. If specified, the document will be highlighted before rendering. |
    | `template`     | `str` or `dict`  | `""`     | A template string or JSON object for the document. This template will be processed by substituting variables and results from queries. |
    | `questions`    | `dict`              | `{}`     | (Optional) A dictionary of natural language questions to be processed by the NL2SQL agent. Each key is a question name, and the value is the question template string. |
    | `queries`      | `dict`              | `{}`     | (Optional) A dictionary of SQL queries to be executed by the QUERYEXECUTOR agent. Each key is a query name, and the value is the SQL query template string. |

    Inputs: 
    - `DEFAULT`: Accepts user input text to initiate document generation.

    Outputs:
    - `DEFAULT`: Outputs the generated document, optionally highlighted, tagged as DOC.

    """

    def __init__(self, **kwargs):
        if "name" not in kwargs:
            kwargs["name"] = "DOCUMENTER"
        super().__init__(**kwargs)

    def _initialize_properties(self):
        super()._initialize_properties()

    ####### inputs / outputs
    def _initialize_inputs(self):
        """Initialize input parameters for the documenter agent. No inputs by default."""
        return

    def _initialize_outputs(self):
        """Initialize outputs for the documenter agent, tagged as DOC"""
        self.add_output("DEFAULT", description="document", tags=["DOC"])

    def issue_nl_query(self, question, progress_id=None, name=None, worker=None, to_param_prefix="QUESTION_RESULTS_"):
        """Issue a natural language query to the NL2SQL agent and route the results back to this agent.

        Parameters:
           question: The natural language question to be processed.
           progress_id: Optional progress identifier for tracking the query progress. Defaults to None.
           name: Optional name for the question, used for routing results. Defaults to None.
           worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
           to_param_prefix: Prefix for the output parameter name where results will be sent. Defaults to "QUESTION_RESULTS_".
        """
        if worker == None:
            worker = self.create_worker(None)

        if progress_id is None:
            progress_id = worker.sid

        # progress
        worker.write_progress(progress_id=progress_id, label='Issuing question:' + question, value=self.current_step / self.num_steps)

        # plan
        p = AgenticPlan(scope=worker.prefix)
        # set input
        p.define_input(name, value=question)
        # set plan
        p.connect_input_to_agent(from_input=name, to_agent="NL2SQL")
        p.connect_agent_to_agent(from_agent="NL2SQL", to_agent=self.name, to_agent_input=to_param_prefix + name)

        # submit plan
        p.submit(worker)

    def issue_sql_query(self, query, progress_id=None, name=None, worker=None, to_param_prefix="QUERY_RESULTS_"):
        """Issue a SQL query to the QUERYEXECUTOR agent and route the results back to this agent.

        Parameters:
           query: The SQL query to be executed.
           progress_id: Optional progress identifier for tracking the query progress. Defaults to None.
           name: Optional name for the query, used for routing results. Defaults to None.
           worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
           to_param_prefix: Prefix for the output parameter name where results will be sent. Defaults to "QUERY_RESULTS_".
        """
        if worker == None:
            worker = self.create_worker(None)

        if progress_id is None:
            progress_id = worker.sid

        # progress
        worker.write_progress(progress_id=progress_id, label='Issuing query:' + query, value=self.current_step / self.num_steps)

        # plan
        p = AgenticPlan(scope=worker.prefix)
        # set input
        p.define_input(name, value=query)
        # set plan
        p.connect_input_to_agent(from_input=name, to_agent="QUERYEXECUTOR")
        p.connect_agent_to_agent(from_agent="QUERYEXECUTOR", to_agent=self.name, to_agent_input=to_param_prefix + name)

        # submit plan
        p.submit(worker)

    def hilite_doc(self, doc, progress_id=None, properties=None, worker=None):
        """Optionally highlight the document using the HILITER agent if 'hilite' property is specified.

        Parameters:
            doc: The document content to be highlighted.
            progress_id: Optional progress identifier for tracking the highlighting progress. Defaults to None.
            properties: Additional properties for processing. Defaults to None.
            worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
        """
        if 'hilite' in properties:
            hilite = properties['hilite']

            if worker == None:
                worker = self.create_worker(None)

            if progress_id is None:
                progress_id = worker.sid

            if properties is None:
                properties = self.properties

            # progress
            worker.write_progress(progress_id=progress_id, label='Highlighting document...', value=self.current_step / self.num_steps)

            session_data = worker.get_all_session_data()

            if session_data is None:
                session_data = {}

            processed_hilite = string_utils.safe_substitute(hilite, **properties, **session_data, **self.results)

            hilite_contents = {"hilite": processed_hilite, "doc": doc}

            hilite_contents_json = json.dumps(hilite_contents, indent=3)

            # plan
            p = AgenticPlan(scope=worker.prefix)
            # set input
            p.define_input("doc", value=hilite_contents_json)
            # set plan
            p.connect_input_to_agent(from_input="doc", to_agent="OPENAI___HILITER")
            p.connect_agent_to_agent(from_agent="OPENAI___HILITER", to_agent=self.name, to_agent_input="DOC")

            # submit plan
            p.submit(worker)

    def process_doc(self, progress_id=None, properties=None, input="", worker=None):
        """Process the document by substituting the template with gathered results and rendering it.

        Parameters:
            progress_id: Optional progress identifier for tracking the processing progress. Defaults to None.
            properties: Additional properties for processing. Defaults to None.
            input: The input text to be included in the document. Defaults to an empty string.
            worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
        """
        if worker == None:
            worker = self.create_worker(None)

        if progress_id is None:
            progress_id = worker.sid

        if properties is None:
            properties = self.properties

        # progress
        worker.write_progress(progress_id=progress_id, label='Processing document...', value=self.current_step / self.num_steps)

        doc = self.substitute_doc(worker, self.results, properties, input)

        if 'hilite' in properties:
            self.hilite_doc(doc, properties=properties, worker=worker, progress_id=progress_id)
        else:
            self.render_doc(doc, properties=properties, worker=worker, progress_id=progress_id)

    def substitute_doc(self, worker, results, properties, input):
        """Substitute the document template with gathered results and contextual information.

        Parameters:
            worker: The worker handling the processing.
            results: Dictionary containing results from queries and questions.
            properties: Additional properties for processing.
            input: The input text to be included in the document.
        Returns:
            The processed document with all substitutions made.
        """
        session_data = worker.get_all_session_data()
        if session_data is None:
            session_data = {}

        template = properties['template']
        if type(template) is dict:
            template = json.dumps(template)

        processed_template = string_utils.safe_substitute(template, **properties, **session_data, **results, input=input)

        return processed_template

    def render_doc(self, doc, progress_id=None, properties=None, worker=None):
        """Render the document by creating a form and sending it to the output stream.

        Parameters:
            doc: The document content to be rendered.
            progress_id: Optional progress identifier for tracking the rendering progress. Defaults to None.
            properties: Additional properties for processing. Defaults to None.
            worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
        """
        if worker == None:
            worker = self.create_worker(None)

        if progress_id is None:
            progress_id = worker.sid

        if properties is None:
            properties = self.properties

        doc_form = build_doc_form(doc)

        # write vis
        worker.write_control(ControlCode.CREATE_FORM, doc_form, output="DOC")

        # progress, done
        worker.write_progress(progress_id=progress_id, label='Done...', value=1.0)

    def default_processor(self, message, input="DEFAULT", properties=None, worker=None):
        """Process messages for the documenter agent, handling user input and query results to generate and render documents.

        Parameters:
            message: The message to process.
            input: The input stream label.
            properties: Additional properties for processing.
            worker: The worker handling the processing.

        Returns:
            None or a response message.
        """
        ##### Upon USER input text
        if input == "DEFAULT":
            if message.isEOS():
                # get all data received from user stream
                stream = message.getStream()

                self.progress_id = stream

                stream_data = worker.get_data(stream)
                input_data = " ".join(stream_data)
                worker.set_data("input", input_data)

                if worker:
                    session_data = worker.get_all_session_data()

                    if session_data is None:
                        session_data = {}

                    # user initiated summarizer, kick off queries from template
                    self.results = {}
                    self.todos = set()

                    self.num_steps = 1
                    if 'hilite' in self.properties:
                        self.num_steps = self.num_steps + 1
                    self.current_step = 0

                    if 'questions' in self.properties:
                        self.num_steps = self.num_steps + len(self.properties['questions'].keys())
                    if 'queries' in self.properties:
                        self.num_steps = self.num_steps + len(self.properties['queries'].keys())

                    # nl questions
                    if 'questions' in self.properties:
                        questions = self.properties['questions']
                        for question_name in questions:
                            q = questions[question_name]
                            question = string_utils.safe_substitute(q, **self.properties, **session_data, input=input_data)
                            self.todos.add(question_name)
                            self.issue_nl_query(question, name=question_name, worker=worker, progress_id=self.progress_id)
                    # db queries
                    if 'queries' in self.properties:
                        queries = self.properties['queries']
                        for query_name in queries:
                            q = queries[query_name]
                            if type(q) == dict:
                                q = json.dumps(q)
                            else:
                                q = str(q)
                            query = string_utils.safe_substitute(q, **self.properties, **session_data, input=input_data)
                            self.todos.add(query_name)
                            self.issue_sql_query(query, name=query_name, worker=worker, progress_id=self.progress_id)
                    if 'questions' not in self.properties and 'queries' not in self.properties:
                        self.process_doc(properties=properties, input=input_data, worker=None, progress_id=self.progress_id)

                    return

            elif message.isBOS():
                stream = message.getStream()

                # init private stream data to empty array
                if worker:
                    worker.set_data(stream, [])
                pass
            elif message.isData():
                # store data value
                data = message.getData()
                stream = message.getStream()

                # append to private stream data
                if worker:
                    worker.append_data(stream, data)

        elif input.find("QUERY_RESULTS_") == 0:
            if message.isData():
                stream = message.getStream()

                # get query
                query = input[len("QUERY_RESULTS_") :]

                data = message.getData()

                if 'result' in data:
                    query_results = data['result']

                    self.results[query] = query_results
                    self.todos.remove(query)

                    # progress
                    self.current_step = len(self.results)
                    q = ""
                    if 'query' in data and data['query']:
                        q = data['query']

                    worker.write_progress(progress_id=self.progress_id, label='Received query results: ' + q, value=self.current_step / self.num_steps)

                    if len(self.todos) == 0:
                        input_data = worker.get_data("input")
                        if input_data is None:
                            input_data = ""
                        self.process_doc(properties=properties, input=input_data, worker=worker, progress_id=self.progress_id)
                else:
                    self.logger.info("nothing found")
        elif input.find("QUESTION_RESULTS_") == 0:
            if message.isData():
                stream = message.getStream()

                # get question
                question = input[len("QUESTION_RESULTS_") :]

                data = message.getData()

                if 'result' in data:
                    question_results = data['result']

                    self.results[question] = question_results
                    self.todos.remove(question)

                    input_data = worker.get_data("input")
                    if input_data is None:
                        input_data = ""
                    # progress
                    self.current_step = len(self.results)
                    q = ""
                    if 'question' in data and data['question']:
                        q = data['question']

                    worker.write_progress(progress_id=self.progress_id, label='Received question results: ' + q, value=self.current_step / self.num_steps)

                    if len(self.todos) == 0:
                        input_data = worker.get_data("input")
                        if input_data is None:
                            input_data = ""
                        self.process_doc(properties=properties, input=input_data, worker=worker, progress_id=self.progress_id)
                else:
                    self.logger.info("nothing found")
        elif input == "DOC":
            if message.isData():
                data = message.getData()

                # progress
                self.current_step = self.num_steps - 1
                worker.write_progress(progress_id=self.progress_id, label='Received highlighted document...', value=self.current_step / self.num_steps)

                doc = str(data)
                self.render_doc(doc, properties=properties, worker=worker, progress_id=self.progress_id)

default_processor(message, input='DEFAULT', properties=None, worker=None)

Process messages for the documenter agent, handling user input and query results to generate and render documents.

Parameters:

Name Type Description Default
message

The message to process.

required
input

The input stream label.

'DEFAULT'
properties

Additional properties for processing.

None
worker

The worker handling the processing.

None

Returns:

Type Description

None or a response message.

Source code in blue/agents/documenter.py
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
def default_processor(self, message, input="DEFAULT", properties=None, worker=None):
    """Process messages for the documenter agent, handling user input and query results to generate and render documents.

    Parameters:
        message: The message to process.
        input: The input stream label.
        properties: Additional properties for processing.
        worker: The worker handling the processing.

    Returns:
        None or a response message.
    """
    ##### Upon USER input text
    if input == "DEFAULT":
        if message.isEOS():
            # get all data received from user stream
            stream = message.getStream()

            self.progress_id = stream

            stream_data = worker.get_data(stream)
            input_data = " ".join(stream_data)
            worker.set_data("input", input_data)

            if worker:
                session_data = worker.get_all_session_data()

                if session_data is None:
                    session_data = {}

                # user initiated summarizer, kick off queries from template
                self.results = {}
                self.todos = set()

                self.num_steps = 1
                if 'hilite' in self.properties:
                    self.num_steps = self.num_steps + 1
                self.current_step = 0

                if 'questions' in self.properties:
                    self.num_steps = self.num_steps + len(self.properties['questions'].keys())
                if 'queries' in self.properties:
                    self.num_steps = self.num_steps + len(self.properties['queries'].keys())

                # nl questions
                if 'questions' in self.properties:
                    questions = self.properties['questions']
                    for question_name in questions:
                        q = questions[question_name]
                        question = string_utils.safe_substitute(q, **self.properties, **session_data, input=input_data)
                        self.todos.add(question_name)
                        self.issue_nl_query(question, name=question_name, worker=worker, progress_id=self.progress_id)
                # db queries
                if 'queries' in self.properties:
                    queries = self.properties['queries']
                    for query_name in queries:
                        q = queries[query_name]
                        if type(q) == dict:
                            q = json.dumps(q)
                        else:
                            q = str(q)
                        query = string_utils.safe_substitute(q, **self.properties, **session_data, input=input_data)
                        self.todos.add(query_name)
                        self.issue_sql_query(query, name=query_name, worker=worker, progress_id=self.progress_id)
                if 'questions' not in self.properties and 'queries' not in self.properties:
                    self.process_doc(properties=properties, input=input_data, worker=None, progress_id=self.progress_id)

                return

        elif message.isBOS():
            stream = message.getStream()

            # init private stream data to empty array
            if worker:
                worker.set_data(stream, [])
            pass
        elif message.isData():
            # store data value
            data = message.getData()
            stream = message.getStream()

            # append to private stream data
            if worker:
                worker.append_data(stream, data)

    elif input.find("QUERY_RESULTS_") == 0:
        if message.isData():
            stream = message.getStream()

            # get query
            query = input[len("QUERY_RESULTS_") :]

            data = message.getData()

            if 'result' in data:
                query_results = data['result']

                self.results[query] = query_results
                self.todos.remove(query)

                # progress
                self.current_step = len(self.results)
                q = ""
                if 'query' in data and data['query']:
                    q = data['query']

                worker.write_progress(progress_id=self.progress_id, label='Received query results: ' + q, value=self.current_step / self.num_steps)

                if len(self.todos) == 0:
                    input_data = worker.get_data("input")
                    if input_data is None:
                        input_data = ""
                    self.process_doc(properties=properties, input=input_data, worker=worker, progress_id=self.progress_id)
            else:
                self.logger.info("nothing found")
    elif input.find("QUESTION_RESULTS_") == 0:
        if message.isData():
            stream = message.getStream()

            # get question
            question = input[len("QUESTION_RESULTS_") :]

            data = message.getData()

            if 'result' in data:
                question_results = data['result']

                self.results[question] = question_results
                self.todos.remove(question)

                input_data = worker.get_data("input")
                if input_data is None:
                    input_data = ""
                # progress
                self.current_step = len(self.results)
                q = ""
                if 'question' in data and data['question']:
                    q = data['question']

                worker.write_progress(progress_id=self.progress_id, label='Received question results: ' + q, value=self.current_step / self.num_steps)

                if len(self.todos) == 0:
                    input_data = worker.get_data("input")
                    if input_data is None:
                        input_data = ""
                    self.process_doc(properties=properties, input=input_data, worker=worker, progress_id=self.progress_id)
            else:
                self.logger.info("nothing found")
    elif input == "DOC":
        if message.isData():
            data = message.getData()

            # progress
            self.current_step = self.num_steps - 1
            worker.write_progress(progress_id=self.progress_id, label='Received highlighted document...', value=self.current_step / self.num_steps)

            doc = str(data)
            self.render_doc(doc, properties=properties, worker=worker, progress_id=self.progress_id)

hilite_doc(doc, progress_id=None, properties=None, worker=None)

Optionally highlight the document using the HILITER agent if 'hilite' property is specified.

Parameters:

Name Type Description Default
doc

The document content to be highlighted.

required
progress_id

Optional progress identifier for tracking the highlighting progress. Defaults to None.

None
properties

Additional properties for processing. Defaults to None.

None
worker

The worker handling the processing. If None, a new worker is created. Defaults to None.

None
Source code in blue/agents/documenter.py
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
def hilite_doc(self, doc, progress_id=None, properties=None, worker=None):
    """Optionally highlight the document using the HILITER agent if 'hilite' property is specified.

    Parameters:
        doc: The document content to be highlighted.
        progress_id: Optional progress identifier for tracking the highlighting progress. Defaults to None.
        properties: Additional properties for processing. Defaults to None.
        worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
    """
    if 'hilite' in properties:
        hilite = properties['hilite']

        if worker == None:
            worker = self.create_worker(None)

        if progress_id is None:
            progress_id = worker.sid

        if properties is None:
            properties = self.properties

        # progress
        worker.write_progress(progress_id=progress_id, label='Highlighting document...', value=self.current_step / self.num_steps)

        session_data = worker.get_all_session_data()

        if session_data is None:
            session_data = {}

        processed_hilite = string_utils.safe_substitute(hilite, **properties, **session_data, **self.results)

        hilite_contents = {"hilite": processed_hilite, "doc": doc}

        hilite_contents_json = json.dumps(hilite_contents, indent=3)

        # plan
        p = AgenticPlan(scope=worker.prefix)
        # set input
        p.define_input("doc", value=hilite_contents_json)
        # set plan
        p.connect_input_to_agent(from_input="doc", to_agent="OPENAI___HILITER")
        p.connect_agent_to_agent(from_agent="OPENAI___HILITER", to_agent=self.name, to_agent_input="DOC")

        # submit plan
        p.submit(worker)

issue_nl_query(question, progress_id=None, name=None, worker=None, to_param_prefix='QUESTION_RESULTS_')

Issue a natural language query to the NL2SQL agent and route the results back to this agent.

Parameters:

Name Type Description Default
question

The natural language question to be processed.

required
progress_id

Optional progress identifier for tracking the query progress. Defaults to None.

None
name

Optional name for the question, used for routing results. Defaults to None.

None
worker

The worker handling the processing. If None, a new worker is created. Defaults to None.

None
to_param_prefix

Prefix for the output parameter name where results will be sent. Defaults to "QUESTION_RESULTS_".

'QUESTION_RESULTS_'
Source code in blue/agents/documenter.py
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
def issue_nl_query(self, question, progress_id=None, name=None, worker=None, to_param_prefix="QUESTION_RESULTS_"):
    """Issue a natural language query to the NL2SQL agent and route the results back to this agent.

    Parameters:
       question: The natural language question to be processed.
       progress_id: Optional progress identifier for tracking the query progress. Defaults to None.
       name: Optional name for the question, used for routing results. Defaults to None.
       worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
       to_param_prefix: Prefix for the output parameter name where results will be sent. Defaults to "QUESTION_RESULTS_".
    """
    if worker == None:
        worker = self.create_worker(None)

    if progress_id is None:
        progress_id = worker.sid

    # progress
    worker.write_progress(progress_id=progress_id, label='Issuing question:' + question, value=self.current_step / self.num_steps)

    # plan
    p = AgenticPlan(scope=worker.prefix)
    # set input
    p.define_input(name, value=question)
    # set plan
    p.connect_input_to_agent(from_input=name, to_agent="NL2SQL")
    p.connect_agent_to_agent(from_agent="NL2SQL", to_agent=self.name, to_agent_input=to_param_prefix + name)

    # submit plan
    p.submit(worker)

issue_sql_query(query, progress_id=None, name=None, worker=None, to_param_prefix='QUERY_RESULTS_')

Issue a SQL query to the QUERYEXECUTOR agent and route the results back to this agent.

Parameters:

Name Type Description Default
query

The SQL query to be executed.

required
progress_id

Optional progress identifier for tracking the query progress. Defaults to None.

None
name

Optional name for the query, used for routing results. Defaults to None.

None
worker

The worker handling the processing. If None, a new worker is created. Defaults to None.

None
to_param_prefix

Prefix for the output parameter name where results will be sent. Defaults to "QUERY_RESULTS_".

'QUERY_RESULTS_'
Source code in blue/agents/documenter.py
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
def issue_sql_query(self, query, progress_id=None, name=None, worker=None, to_param_prefix="QUERY_RESULTS_"):
    """Issue a SQL query to the QUERYEXECUTOR agent and route the results back to this agent.

    Parameters:
       query: The SQL query to be executed.
       progress_id: Optional progress identifier for tracking the query progress. Defaults to None.
       name: Optional name for the query, used for routing results. Defaults to None.
       worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
       to_param_prefix: Prefix for the output parameter name where results will be sent. Defaults to "QUERY_RESULTS_".
    """
    if worker == None:
        worker = self.create_worker(None)

    if progress_id is None:
        progress_id = worker.sid

    # progress
    worker.write_progress(progress_id=progress_id, label='Issuing query:' + query, value=self.current_step / self.num_steps)

    # plan
    p = AgenticPlan(scope=worker.prefix)
    # set input
    p.define_input(name, value=query)
    # set plan
    p.connect_input_to_agent(from_input=name, to_agent="QUERYEXECUTOR")
    p.connect_agent_to_agent(from_agent="QUERYEXECUTOR", to_agent=self.name, to_agent_input=to_param_prefix + name)

    # submit plan
    p.submit(worker)

process_doc(progress_id=None, properties=None, input='', worker=None)

Process the document by substituting the template with gathered results and rendering it.

Parameters:

Name Type Description Default
progress_id

Optional progress identifier for tracking the processing progress. Defaults to None.

None
properties

Additional properties for processing. Defaults to None.

None
input

The input text to be included in the document. Defaults to an empty string.

''
worker

The worker handling the processing. If None, a new worker is created. Defaults to None.

None
Source code in blue/agents/documenter.py
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
def process_doc(self, progress_id=None, properties=None, input="", worker=None):
    """Process the document by substituting the template with gathered results and rendering it.

    Parameters:
        progress_id: Optional progress identifier for tracking the processing progress. Defaults to None.
        properties: Additional properties for processing. Defaults to None.
        input: The input text to be included in the document. Defaults to an empty string.
        worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
    """
    if worker == None:
        worker = self.create_worker(None)

    if progress_id is None:
        progress_id = worker.sid

    if properties is None:
        properties = self.properties

    # progress
    worker.write_progress(progress_id=progress_id, label='Processing document...', value=self.current_step / self.num_steps)

    doc = self.substitute_doc(worker, self.results, properties, input)

    if 'hilite' in properties:
        self.hilite_doc(doc, properties=properties, worker=worker, progress_id=progress_id)
    else:
        self.render_doc(doc, properties=properties, worker=worker, progress_id=progress_id)

render_doc(doc, progress_id=None, properties=None, worker=None)

Render the document by creating a form and sending it to the output stream.

Parameters:

Name Type Description Default
doc

The document content to be rendered.

required
progress_id

Optional progress identifier for tracking the rendering progress. Defaults to None.

None
properties

Additional properties for processing. Defaults to None.

None
worker

The worker handling the processing. If None, a new worker is created. Defaults to None.

None
Source code in blue/agents/documenter.py
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
def render_doc(self, doc, progress_id=None, properties=None, worker=None):
    """Render the document by creating a form and sending it to the output stream.

    Parameters:
        doc: The document content to be rendered.
        progress_id: Optional progress identifier for tracking the rendering progress. Defaults to None.
        properties: Additional properties for processing. Defaults to None.
        worker: The worker handling the processing. If None, a new worker is created. Defaults to None.
    """
    if worker == None:
        worker = self.create_worker(None)

    if progress_id is None:
        progress_id = worker.sid

    if properties is None:
        properties = self.properties

    doc_form = build_doc_form(doc)

    # write vis
    worker.write_control(ControlCode.CREATE_FORM, doc_form, output="DOC")

    # progress, done
    worker.write_progress(progress_id=progress_id, label='Done...', value=1.0)

substitute_doc(worker, results, properties, input)

Substitute the document template with gathered results and contextual information.

Parameters:

Name Type Description Default
worker

The worker handling the processing.

required
results

Dictionary containing results from queries and questions.

required
properties

Additional properties for processing.

required
input

The input text to be included in the document.

required

Returns: The processed document with all substitutions made.

Source code in blue/agents/documenter.py
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
def substitute_doc(self, worker, results, properties, input):
    """Substitute the document template with gathered results and contextual information.

    Parameters:
        worker: The worker handling the processing.
        results: Dictionary containing results from queries and questions.
        properties: Additional properties for processing.
        input: The input text to be included in the document.
    Returns:
        The processed document with all substitutions made.
    """
    session_data = worker.get_all_session_data()
    if session_data is None:
        session_data = {}

    template = properties['template']
    if type(template) is dict:
        template = json.dumps(template)

    processed_template = string_utils.safe_substitute(template, **properties, **session_data, **results, input=input)

    return processed_template
Last update: 2025-10-09