| |
| |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| -module(hk_schema). |
| -include("hyperkitty.hrl"). |
|
|
| -export([to_map/1]). |
|
|
| to_map(#agent{} = A) -> |
| #{ |
| agent_id => A#agent.agent_id, |
| objective => opt_map(A#agent.objective), |
| state => A#agent.state, |
| capabilities => A#agent.capabilities, |
| action_count => A#agent.action_count, |
| started_at_ms => A#agent.started_at_ms, |
| updated_at_ms => A#agent.updated_at_ms, |
| limits => A#agent.limits, |
| result => term_or_null(A#agent.result), |
| failure_reason => term_or_null(A#agent.failure_reason) |
| }; |
| to_map(#agent_objective{} = O) -> |
| #{ |
| objective_id => O#agent_objective.objective_id, |
| agent_id => O#agent_objective.agent_id, |
| description => O#agent_objective.description, |
| success_criteria => bin_or_null(O#agent_objective.success_criteria), |
| created_at_ms => O#agent_objective.created_at_ms |
| }; |
| to_map(#browser_session{} = S) -> |
| #{ |
| session_id => S#browser_session.session_id, |
| owner_agent_id => bin_or_null(S#browser_session.owner_agent_id), |
| profile => S#browser_session.profile, |
| status => S#browser_session.status, |
| tabs => S#browser_session.tabs, |
| created_at_ms => S#browser_session.created_at_ms, |
| updated_at_ms => S#browser_session.updated_at_ms |
| }; |
| to_map(#browser_tab{} = T) -> |
| #{ |
| tab_id => T#browser_tab.tab_id, |
| session_id => T#browser_tab.session_id, |
| url => bin_or_null(T#browser_tab.url), |
| title => bin_or_null(T#browser_tab.title), |
| status => T#browser_tab.status, |
| created_at_ms => T#browser_tab.created_at_ms, |
| updated_at_ms => T#browser_tab.updated_at_ms |
| }; |
| to_map(#search_job{} = J) -> |
| #{ |
| job_id => J#search_job.job_id, |
| owner_agent_id => bin_or_null(J#search_job.owner_agent_id), |
| query => J#search_job.query, |
| stage => J#search_job.stage, |
| max_iterations => J#search_job.max_iterations, |
| iteration => J#search_job.iteration, |
| results => J#search_job.results, |
| created_at_ms => J#search_job.created_at_ms, |
| updated_at_ms => J#search_job.updated_at_ms |
| }; |
| to_map(#search_result{} = R) -> |
| #{ |
| result_id => R#search_result.result_id, |
| job_id => R#search_result.job_id, |
| url => R#search_result.url, |
| normalized_url => R#search_result.normalized_url, |
| title => bin_or_null(R#search_result.title), |
| snippet => bin_or_null(R#search_result.snippet), |
| rank => term_or_null(R#search_result.rank), |
| source_document_id => bin_or_null(R#search_result.source_document_id), |
| created_at_ms => R#search_result.created_at_ms |
| }; |
| to_map(#source_document{} = D) -> |
| #{ |
| document_id => D#source_document.document_id, |
| url => D#source_document.url, |
| fetched_at_ms => D#source_document.fetched_at_ms, |
| content_type => bin_or_null(D#source_document.content_type), |
| extracted_text => bin_or_null(D#source_document.extracted_text), |
| citation_map => D#source_document.citation_map |
| }; |
| to_map(#message{} = M) -> |
| #{ |
| message_id => M#message.message_id, |
| sender => M#message.sender, |
| recipient => M#message.recipient, |
| timestamp => M#message.timestamp, |
| conversation_id => M#message.conversation_id, |
| message_type => M#message.message_type, |
| payload => M#message.payload, |
| status => M#message.status |
| }; |
| to_map(#operation{} = Op) -> |
| #{ |
| operation_id => Op#operation.operation_id, |
| request_id => Op#operation.request_id, |
| kind => Op#operation.kind, |
| status => Op#operation.status, |
| subject => case Op#operation.subject of |
| {Type, Id} -> #{type => Type, id => Id}; |
| undefined -> null |
| end, |
| error => term_or_null(Op#operation.error), |
| created_at_ms => Op#operation.created_at_ms, |
| updated_at_ms => Op#operation.updated_at_ms |
| }. |
|
|
| |
|
|
| opt_map(undefined) -> null; |
| opt_map(Record) -> to_map(Record). |
|
|
| bin_or_null(undefined) -> null; |
| bin_or_null(B) -> B. |
|
|
| term_or_null(undefined) -> null; |
| term_or_null(T) -> T. |
|
|
