tg-agent 1.1.8 → 1.1.9
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.
- package/dist/index.js +14 -14
- package/package.json +1 -1
package/dist/index.js
CHANGED
|
@@ -1,17 +1,17 @@
|
|
|
1
|
-
import
|
|
2
|
-
`)}function
|
|
3
|
-
`)}function
|
|
4
|
-
`),g)}}const
|
|
1
|
+
import N from"node:path";import X from"node:fs/promises";import ye from"node-telegram-bot-api";import{discoverAuthStorage as he,discoverModels as Me}from"@mariozechner/pi-coding-agent";import{getOAuthProviders as ve}from"@mariozechner/pi-ai";import{config as m,assertConfig as Se}from"./config.js";import{resolveApiKeyForProvider as ke,resolveProxyInfo as _e}from"./auth.js";import{applyFetchProxy as Ae}from"./proxy.js";import{runPiAgentPrompt as Pe}from"./piAgentRunner.js";import{readCodexOAuth as xe}from"./codexAuth.js";import{formatMcpTarget as Ce,freezeMcpCatalog as J,getMcpCatalog as Z,loadMcpServers as be,loadMcpServersSync as Ee,probeMcpServer as Re,refreshMcpCatalog as De}from"./mcp.js";import{appendMessage as Y,closeSession as ze,createSession as F,deleteSessionFile as j,getActiveSession as z,listSessions as T,loadChatState as I,pruneExpiredSessions as ee,resetSession as Le,saveChatState as y,setActiveSession as Ue,setWorkspaceDir as Ne}from"./sessionStore.js";import{chunkText as je,createQueueMap as We,createSemaphore as Oe,ensureDir as K,expandHome as Be,nowMs as C,shortId as te}from"./utils.js";Se();const h=new ye(m.telegramToken,{polling:!0}),ne=We(),Fe=Oe(m.maxConcurrent),G=new Map;function Ke(e,t){G.set(e,t)}function Ge(e){G.delete(e)}function Qe(e){const t=e.trim();if(t){if(t==="Markdown"||t==="MarkdownV2"||t==="HTML")return t;console.warn(`[tg-agent] invalid TELEGRAM_PARSE_MODE=${t}, ignoring`)}}const Ve="image/jpeg";function qe(){return N.join(m.agentDir,"uploads")}function He(e){return e.length===0?null:e.reduce((t,n)=>{const o=t.file_size??t.width*t.height;return(n.file_size??n.width*n.height)>o?n:t})}function Xe(e){return!!e?.startsWith("image/")}function Je(e){switch(N.extname(e).toLowerCase()){case".png":return"image/png";case".jpg":case".jpeg":return"image/jpeg";case".webp":return"image/webp";case".gif":return"image/gif";default:return}}function Ze(e){if(!Number.isFinite(e)||e<=0)return"0 B";const t=["B","KB","MB","GB"];let n=e,o=0;for(;n>=1024&&o<t.length-1;)n/=1024,o+=1;return`${n.toFixed(n>=10||o===0?0:1)} ${t[o]}`}function Ye(e){return e.length===0?"":e.map(t=>{const n=[];t.mimeType&&n.push(t.mimeType),t.bytes>0&&n.push(Ze(t.bytes));const o=n.length>0?` (${n.join(", ")})`:"";return`Attachment saved: ${t.path}${o}`}).join(`
|
|
2
|
+
`)}function Te(e){const t=[];if(e.photo&&e.photo.length>0){const n=He(e.photo);n&&t.push({kind:"photo",fileId:n.file_id,fileSize:n.file_size})}return e.document&&t.push({kind:"document",fileId:e.document.file_id,fileName:e.document.file_name,mimeType:e.document.mime_type,fileSize:e.document.file_size}),t}async function Ie(e){if(e.length===0)return{resolved:[],images:[]};const t=qe();await K(t);const n=[],o=[];for(const r of e)try{const c=await h.downloadFile(r.fileId,t),p=await X.stat(c),a=r.fileName??N.basename(c),s=r.mimeType??Je(c),d=r.kind==="photo"||Xe(s),i={path:c,name:a,mimeType:s,bytes:p.size,isImage:d};if(n.push(i),d){const u=await X.readFile(c);o.push({type:"image",data:u.toString("base64"),mimeType:s??Ve})}}catch(c){console.warn("[tg-agent] attachment download failed",c)}return{resolved:n,images:o}}const et=new Set(["_","*","[","]","(",")","~","`",">","#","+","-","=","|","{","}",".","!"]);function L(e,t){let n=0;for(let o=0;o<e.length;o+=1){const r=e[o];if(r==="\\"){o+=1;continue}r===t&&(n+=1)}return n}function tt(e){if(L(e,"*")%2!==0||L(e,"_")%2!==0||L(e,"`")%2!==0)return!1;const r=L(e,"["),c=L(e,"]");return r===c}function nt(e){for(let t=0;t<e.length;t+=1){const n=e[t];if(n==="\\"){t+=1;continue}if(et.has(n))return!1}return!0}function oe(e){const t=Qe(m.telegramParseMode);return t?[t]:nt(e)?["MarkdownV2","Markdown"]:tt(e)?["Markdown"]:[]}const ot=600*1e3,A=new Map,Q=new Set,rt=new Map([["codex","openai-codex"],["antigravity","google-antigravity"],["gemini","google-gemini-cli"],["gemini-cli","google-gemini-cli"]]),it={"codex-mini-latest":"gpt-5.1-codex-mini","codex-max-latest":"gpt-5.1-codex-max","codex-latest":"gpt-5.2-codex"};function E(e){const t=e.trim().toLowerCase();return rt.get(t)??t}function st(e,t){return e!=="openai-codex"?t:it[t]??t}async function R(){await K(m.agentDir);const e=he(m.agentDir),t=xe();t&&e.setRuntimeApiKey("openai-codex",t.accessToken);const n=Me(e,m.agentDir);return{authStorage:e,modelRegistry:n}}function re(e){const t=A.get(e);t&&(clearTimeout(t.timeoutId),A.delete(e))}function at(e,t){const n=A.get(e);n&&(clearTimeout(n.timeoutId),A.delete(e),n.reject(new Error(t)))}async function ie(e,t,n,o){if(A.has(e))throw new Error("Login is already awaiting input.");const r=n.placeholder?`${n.message} (${n.placeholder})`:n.message;return await l(t,r,o),await new Promise((c,p)=>{const a=setTimeout(()=>{A.delete(e),p(new Error("Login prompt timed out."))},ot);A.set(e,{resolve:c,reject:p,timeoutId:a,chatId:t})})}async function se(e){const t=String(e),n=G.get(t);if(!n){await l(e,"No active request to stop.");return}n.cancelRequested=!0,n.abortRequested=!0,n.status&&n.status.update("Cancelled by user.",!0),n.abort&&n.abort(),(!n.status||n.chatId!==e)&&await l(e,"Stopping current request...")}function ct(){console.log(`[tg-agent] modelProvider=${m.modelProvider} modelRef=${m.modelRef} sessionDir=${m.sessionDir} maxConcurrent=${m.maxConcurrent}`),console.log(`[tg-agent] agentDir=${m.agentDir} workspaceDir=${m.workspaceDir}`);const e=Ee(m.agentDir),t=e.length>0?`on (${e.length})`:"off";console.log(`[tg-agent] tools fetchMaxBytes=${m.fetchMaxBytes} fetchTimeoutMs=${m.fetchTimeoutMs} mcp=${t}`),console.log(`[tg-agent] proxy url=${m.proxyUrl||"(none)"} fetchProxy=${m.fetchProxyUrl||"(none)"}`),console.log(`[tg-agent] tls extra_ca=${m.tlsExtraCaCerts||"(empty)"} reject_unauthorized=${m.tlsRejectUnauthorized===null?"(unset)":String(m.tlsRejectUnauthorized)}`);try{const{source:o}=ke(m.modelProvider);console.log(`[tg-agent] authSource=${o}`)}catch(o){const r=o instanceof Error?o.message:String(o);console.warn(`[tg-agent] authSource=missing (${r})`)}const n=_e();console.log(n?`[tg-agent] proxyUrl=${n.url} kind=${n.kind} source=${n.source}`:"[tg-agent] proxyUrl=(none)")}ct(),K(m.agentDir).catch(e=>{console.warn(`[tg-agent] ensure agentDir failed: ${e instanceof Error?e.message:String(e)}`)}),(async()=>{try{await Z(m.agentDir,{timeoutMs:3e3,maxBytes:m.fetchMaxBytes,maxChars:2e3}),J()}catch(e){const t=e instanceof Error?e.message:String(e);console.warn(`[tg-agent] mcp catalog warmup failed: ${t}`)}})();const W=Ae();console.log(W?`[tg-agent] fetchProxy=${W.url} kind=${W.kind} source=${W.source}`:"[tg-agent] fetchProxy=(none)");function V(e){if(!e.startsWith("/"))return null;const t=e.trim(),[n,...o]=t.split(" ");return{command:n.split("@")[0].slice(1).toLowerCase(),args:o.join(" ").trim()}}async function ae(e,t,n){let o=z(e);return o||(o=F(e,""),await y(e),await l(t,`Created session ${o.id}.`,n)),o}function lt(e,t){const n=e.getAll(),o=new Map;for(const r of n){const c=o.get(r.provider)??{count:0,auth:t.hasAuth(r.provider)};c.count+=1,o.set(r.provider,c)}return Array.from(o.entries()).sort((r,c)=>r[0].localeCompare(c[0])).map(([r,c])=>({label:`${r} (${c.count}, ${c.auth?"ok":"no auth"})`,command:`/provider ${r}`}))}function ut(e,t,n=[]){const o=e.getAll().filter(a=>a.provider===t).sort((a,s)=>a.id.localeCompare(s.id));if(n.length===0)return o.map(a=>({label:a.id,command:`/model ${t}/${a.id}`}));const r=new Map(o.map(a=>[a.id,a])),c=new Set,p=[];for(const a of n){const s=r.get(a);!s||c.has(s.id)||(c.add(s.id),p.push(s))}for(const a of o)c.has(a.id)||p.push(a);return p.map(a=>({label:a.id,command:`/model ${t}/${a.id}`}))}function ft(e,t,n=3){const o=new Set,r=[],c=T(e);for(const p of c){if(p.modelProvider!==t)continue;const a=p.modelId;if(!(!a||o.has(a))&&(o.add(a),r.push(a),r.length>=n))break}return r}function dt(e){const t=e.trim();if(!t)return null;if(t.includes("/")){const[n,o]=t.split("/",2);return!n||!o?null:{provider:E(n),modelId:o.trim()}}return{modelId:t}}const mt=6e4,ce=new Map;function le(e,t){const n=Date.now(),o=ce.get(e)??0;n-o<mt||(ce.set(e,n),console.warn(`[tg-agent] unauthorized user=${e} chat=${t}`))}function ue(e){const t=m.telegramAllowedUsers;return!t||t.size===0?!0:t.has(e)}function pt(e){return e.length===0?"No sessions found.":["Sessions:",...e.map(n=>{const o=new Date(n.updatedAt).toISOString();return`- ${n.id} | ${n.title} | updated ${o}`})].join(`
|
|
3
|
+
`)}function O(e,t,n){const o=String(e);return t!==void 0&&!n?null:t!==void 0&&n?`${o}_${t}`:o}function q(e,t){if(t.workspaceDir)return t.workspaceDir;const n=m.workspaceMappings.get(e);return n||m.workspaceDir}async function fe(e,t,n,o){const r={};return n&&(r.parse_mode=n),o!==void 0&&(r.message_thread_id=o),await h.sendMessage(e,t,r)}async function de(e,t,n,o,r){const c={chat_id:e,message_id:t};o&&(c.parse_mode=o),r!==void 0&&(c.message_thread_id=r),await h.editMessageText(n,c)}async function me(e,t){let n;for(const o of t)try{return await e(o)}catch(r){n=r;const c=r instanceof Error?r.message:String(r);console.warn(`[tg-agent] parse_mode ${o} failed, trying fallback: ${c}`)}try{return await e(void 0)}catch(o){if(n){const r=o instanceof Error?o.message:String(o);console.warn(`[tg-agent] parse_mode fallback to plain failed: ${r}`)}throw o}}async function gt(e,t,n){return await me(o=>fe(e,t,o,n),oe(t))}async function wt(e,t,n,o){await me(r=>de(e,t,n,r,o),oe(n))}async function l(e,t,n){const o=je(t,3900);for(const r of o)await gt(e,r,n)}const pe=1800*1e3,ge=2e3,$t=60,S=new Map;function yt(){const e=C()-pe;for(const[o,r]of S)r.createdAt<e&&S.delete(o);if(S.size<=ge)return;const t=Array.from(S.entries()).sort((o,r)=>o[1].createdAt-r[1].createdAt),n=S.size-ge;for(let o=0;o<n;o+=1)S.delete(t[o][0])}function ht(e){yt();let t=te();for(;S.has(t);)t=te();return S.set(t,{command:e,createdAt:C()}),t}function Mt(e){if(!e.startsWith("cmd:"))return null;const t=e.slice(4),n=S.get(t);return n?C()-n.createdAt>pe?(S.delete(t),null):n.command:null}function vt(e,t){if(e.length<=t)return[e];const n=[];for(let o=0;o<e.length;o+=t)n.push(e.slice(o,o+t));return n}function St(e,t){const n=[];let o=[];for(const r of e){const c=ht(r.command);o.push({text:r.label,callback_data:`cmd:${c}`}),o.length>=t&&(n.push(o),o=[])}return o.length>0&&n.push(o),n}async function kt(e,t,n,o){const r={reply_markup:{inline_keyboard:n}};return o!==void 0&&(r.message_thread_id=o),await h.sendMessage(e,t,r)}async function we(e,t,n,o){const r=o?.perRow??1,c=o?.pageSize??$t,p=o?.footer,a=o?.messageThreadId,s=vt(n,c);for(let d=0;d<s.length;d+=1){const i=s[d],u=s.length>1?` (page ${d+1}/${s.length})`:"",f=[`${t}${u}`];p&&f.push(p);const g=St(i,r);await kt(e,f.join(`
|
|
4
|
+
`),g,a)}}const _t=3900,At=1200;function H(e,t=_t){return e.length<=t?{text:e,truncated:!1}:{text:`${e.slice(0,Math.max(0,t-15))}...
|
|
5
5
|
|
|
6
|
-
[truncated]`,truncated:!0}}async function
|
|
7
|
-
args: ${
|
|
8
|
-
`)}async function
|
|
9
|
-
`);await l(e,r);return}case"new":{try{const
|
|
6
|
+
[truncated]`,truncated:!0}}async function Pt(e,t){const n="Working...",r=(await fe(e,n,void 0,t)).message_id;let c=n,p=0,a=Promise.resolve();const s=(i,u=!1)=>{const f=Date.now();!u&&f-p<At||i!==c&&(c=i,p=f,a=a.then(async()=>{await de(e,r,i,void 0,t)}).catch(g=>{console.warn("[tg-agent] status edit failed",g)}))},d=i=>{c=i,p=Date.now(),a=a.then(async()=>{await wt(e,r,i,t)}).catch(u=>{console.warn("[tg-agent] status edit failed",u)})};return{update:(i,u=!1)=>{const f=H(i).text;s(f,u)},finalize:async i=>{const u=H(i).text;d(u),await a},fail:async i=>{const u=H(i).text;d(u),await a}}}function xt(e){switch(e.type){case"agent_start":return"Working...";case"tool_start":return`Running tool: ${e.name}
|
|
7
|
+
args: ${Ct(e.args)}`;case"tool_end":return e.ok?`Tool finished: ${e.name} (${e.durationMs}ms)`:`Tool failed: ${e.name} (${e.durationMs}ms)`;case"message_start":return e.role==="assistant"?"Generating reply...":null;case"message_end":return e.role==="assistant"?"Finalizing reply...":null;case"heartbeat":return`Working... ${Math.max(1,Math.round(e.elapsedMs/1e3))}s`;default:return null}}function Ct(e){try{const t=new WeakSet,n=200,o=12,r=12,c=3,p=(d,i)=>{if(d==null)return d;if(typeof d=="string")return d.length>n?`${d.slice(0,n)}...`:d;if(typeof d=="number"||typeof d=="boolean")return d;if(typeof d=="bigint")return d.toString();if(typeof d=="function")return"[function]";if(typeof d!="object")return String(d);if(t.has(d))return"[circular]";if(i>=c)return"[truncated]";if(t.add(d),Array.isArray(d)){const g=d.slice(0,r).map(w=>p(w,i+1));return d.length>r&&g.push("[truncated]"),g}const u=Object.entries(d),f={};for(const[g,w]of u.slice(0,o))f[g]=p(w,i+1);return u.length>o&&(f._truncated=!0),f},a=p(e,0),s=JSON.stringify(a);return s?s.length>700?`${s.slice(0,700)}...`:s:"null"}catch{return"[unavailable]"}}function B(e){const t=e instanceof Error?e.message:String(e),n=e?.cause;return n instanceof Error?`${t} (${n.message})`:n?`${t} (${String(n)})`:t}function bt(e,t=140){return e.length<=t?e:`${e.slice(0,t-3)}...`}async function Et(){const e=await be(m.agentDir);if(e.length===0)return"No MCP servers configured. Add [mcp_servers.*] to ~/.tg-agent/config.toml.";const t=await Promise.all(e.map(o=>Re(o))),n=["MCP servers:"];for(let o=0;o<e.length;o+=1){const r=e[o],c=t[o],p=c.ok?"ok":`error: ${bt(c.error??"unknown")}`;n.push(`- ${r.name} (${r.type}) ${Ce(r)} status=${p} (${c.durationMs}ms)`)}return n.join(`
|
|
8
|
+
`)}async function $e(e,t,n,o,r){const c=r,a=O(e,c,c!==void 0);if(!a)return;const s=await I(a),d=ee(s);switch(d.length>0&&(await y(s),await Promise.all(d.map(i=>j(a,i).catch(u=>{console.warn(`[tg-agent] cleanup session file failed id=${i}`,u)})))),n){case"start":case"help":{const i=["Commands:","/new [title] - create a new session","/list - list sessions","/use <id> - switch active session","/close [id] - close a session (default: active)","/reset - clear active session history","/workspace [path] - view or set workspace directory","/providers - list available providers","/models [provider] - list models for provider","/provider <name> - set provider for current session","/model <provider>/<model> - set model for current session","/mcp - list configured MCP servers","/mcp refresh - reload MCP catalog","/status - show session and workspace settings","/login <provider> - login to OAuth provider","/logout <provider> - logout from provider","/stop - stop the current running request","","Tips:","Send images or files to attach them to the prompt."].join(`
|
|
9
|
+
`);await l(e,i,r);return}case"new":{try{const u=F(s,o||"");await y(s),await l(e,`Created session ${u.id} (${u.title}).`,r)}catch(i){await l(e,i.message,r)}return}case"list":{await y(s),await l(e,pt(T(s)),r);return}case"use":{if(!o){await l(e,"Usage: /use <id>",r);return}if(!Ue(s,o)){await l(e,`Session not found: ${o}`,r);return}await y(s),await l(e,`Active session set to ${o}.`,r);return}case"close":{const i=o||s.activeSessionId;if(!i){await l(e,"No active session to close.",r);return}if(!ze(s,i)){await l(e,`Session not found: ${i}`,r);return}await y(s),await j(a,i).catch(f=>{console.warn(`[tg-agent] delete session file failed id=${i}`,f)}),await l(e,`Closed session ${i}.`,r);return}case"reset":{const i=z(s);if(!i){await l(e,"No active session.",r);return}Le(i),await y(s),await j(a,i.id).catch(u=>{console.warn(`[tg-agent] reset session file failed id=${i.id}`,u)}),await l(e,`Reset session ${i.id}.`,r);return}case"workspace":{if(!o){const u=q(a,s),f=s.workspaceDir?"chat":m.workspaceMappings.has(a)?"config":"default";await l(e,`Workspace: ${u} (source: ${f})`,r);return}const i=N.resolve(Be(o.trim()));Ne(s,i),await y(s),await l(e,`Workspace set to: ${i}`,r);return}case"providers":{const{authStorage:i,modelRegistry:u}=await R(),f=u.getError(),g=lt(u,i);if(g.length===0){const $=f?`Warning: ${f}
|
|
10
10
|
|
|
11
|
-
No providers found.`:"No providers found.";await l(e
|
|
12
|
-
`),
|
|
13
|
-
`);await l(e
|
|
14
|
-
`));return}const
|
|
15
|
-
`))},onPrompt:
|
|
11
|
+
No providers found.`:"No providers found.";await l(e,$,r);return}const w=[];f&&w.push(`Warning: ${f}`),w.push("Providers:"),await we(e,w.join(`
|
|
12
|
+
`),g,{perRow:1,footer:"Tap a provider to set it. Use /models <provider> to list models.",messageThreadId:r});return}case"models":{const{authStorage:i,modelRegistry:u}=await R();let f=o?E(o):"";if(!f){const P=z(s);if(!P?.modelProvider){await l(e,"Usage: /models <provider>",r);return}f=P.modelProvider}const g=i.hasAuth(f),w=ft(s,f),$=ut(u,f,w);if($.length===0){await l(e,`No models found for provider ${f}.`,r);return}await we(e,`Models for ${f} (auth: ${g?"ok":"missing"}):`,$,{perRow:1,footer:"Tap a model to set it.",messageThreadId:r});return}case"provider":{if(!o){await l(e,"Usage: /provider <name>",r);return}const i=E(o),{modelRegistry:u}=await R();if(!u.getAll().some(w=>w.provider===i)){await l(e,`Unknown provider: ${i}`,r);return}const g=await ae(s,e,r);g.modelProvider=i,g.modelId=void 0,g.updatedAt=C(),await y(s),await l(e,`Session ${g.id} provider set to ${i}.`,r);return}case"model":{const i=dt(o);if(!i){await l(e,"Usage: /model <provider>/<model>",r);return}const u=await ae(s,e,r),f=E(i.provider??u.modelProvider??"");if(!f){await l(e,"Usage: /model <provider>/<model>",r);return}const{modelRegistry:g}=await R(),w=st(f,i.modelId);if(!g.find(f,w)){await l(e,`Model not found: ${f}/${w}`,r);return}u.modelProvider=f,u.modelId=w,u.updatedAt=C(),await y(s),await l(e,`Session ${u.id} model set to ${f}/${w}.`,r);return}case"status":{const i=z(s);if(!i){await l(e,"No active session.",r);return}const u=m.modelRef.includes("/")?m.modelRef:`${m.modelProvider}/${m.modelRef}`,f=i.modelProvider||i.modelId?`Model: ${i.modelProvider??"?"}/${i.modelId??"?"}`:`Model: ${u} (default)`,g=q(a,s),w=s.workspaceDir?"chat":m.workspaceMappings.has(a)?"config":"default",$=[`Session: ${i.id} (${i.title})`,f,`Workspace: ${g} (${w})`].join(`
|
|
13
|
+
`);await l(e,$,r);return}case"mcp":{if(o.trim().toLowerCase()==="refresh"){De(),await Z(m.agentDir,{timeoutMs:5e3,maxBytes:m.fetchMaxBytes,maxChars:3e3}),J(),await l(e,"MCP catalog refreshed.",r);return}const i=await Et();await l(e,i,r);return}case"login":{const i=ve();if(!o){const g=i.map(w=>`- ${w.id} (${w.name})`);await l(e,["OAuth providers:",...g].join(`
|
|
14
|
+
`),r);return}const u=E(o);if(Q.has(t)){await l(e,"Login already in progress.",r);return}const f=i.find(g=>g.id===u);if(!f){await l(e,`Unknown OAuth provider: ${u}`,r);return}Q.add(t);try{const{authStorage:g}=await R();await l(e,`Starting login for ${f.id}...`,r),await g.login(f.id,{onAuth:({url:w,instructions:$})=>{const P=["Open this URL in your browser:",w];$&&P.push("",$),l(e,P.join(`
|
|
15
|
+
`),r)},onPrompt:w=>ie(t,e,w,r),onProgress:w=>{l(e,w,r)},onManualCodeInput:()=>ie(t,e,{message:"Paste the authorization code:"},r)}),await l(e,`Login completed for ${f.id}.`,r)}catch(g){const w=B(g);await l(e,`Login failed: ${w}`,r)}finally{Q.delete(t),re(t)}return}case"logout":{if(!o){await l(e,"Usage: /logout <provider>",r);return}const i=E(o),{authStorage:u}=await R();u.logout(i),await l(e,`Logged out from ${i}.`,r);return}default:await l(e,`Unknown command: ${n}`,r)}}async function Rt(e,t,n,o=[],r){const c=V(n);if(c){await $e(e,t,c.command,c.args,r);return}const a=O(e,r,r!==void 0);if(!a)return;const s=await I(a),d=ee(s);d.length>0&&(await y(s),await Promise.all(d.map(M=>j(a,M).catch(b=>{console.warn(`[tg-agent] cleanup session file failed id=${M}`,b)}))));let i=z(s);if(!i)try{i=F(s,""),await y(s),await l(e,`Created session ${i.id}.`,r)}catch(M){await l(e,M.message,r);return}const{resolved:u,images:f}=await Ie(o);if(o.length>0&&u.length===0&&!n.trim()){await l(e,"Failed to download attachments.",r);return}const g=Ye(u),w=[];n.trim()&&w.push(n.trim()),g&&w.push(g);const $=w.join(`
|
|
16
16
|
|
|
17
|
-
`)||"Attachment received.",
|
|
17
|
+
`)||"Attachment received.",P={role:"user",content:$,ts:C()};Y(i,P,m.maxHistoryMessages),await y(s),console.log(`[tg-agent] request user=${t} chat=${a} session=${i.id} messages=${i.messages.length} textLen=${$.length} attachments=${u.length} provider=${i.modelProvider??"-"} model=${i.modelId??"-"}`);let v=null;try{v=await Pt(e,r)}catch(M){console.warn("[tg-agent] status message failed",M)}const D={sessionId:i.id,chatId:e,messageThreadId:r,status:v,abortRequested:!1,cancelRequested:!1};Ke(a,D);try{const M=await Fe(async()=>{const U=await Pe({chatId:a,sessionId:i.id,prompt:$,images:f,systemPrompt:m.systemPrompt,modelProvider:i.modelProvider,modelId:i.modelId,workspaceDir:q(a,s),telegram:{chatId:e,sendPhoto:async(k,_)=>{const x={};_&&(x.caption=_),r!==void 0&&(x.message_thread_id=r),await h.sendPhoto(e,k,x)},sendDocument:async(k,_)=>{const x={};_&&(x.caption=_),r!==void 0&&(x.message_thread_id=r),await h.sendDocument(e,k,x)}},onAbortReady:k=>{D.abort=k,D.abortRequested&&k()},onStatus:v?k=>{const _=xt(k);_&&v?.update(_)}:void 0});return console.log(`[tg-agent] response session=${i.id} model=${U.modelProvider}/${U.modelId} file=${U.sessionFile}`),U.text});if(D.cancelRequested){v?await v.finalize("Cancelled by user."):await l(e,"Cancelled by user.",r);return}const b={role:"assistant",content:M,ts:C()};Y(i,b,m.maxHistoryMessages),await y(s),v?await v.finalize(M):await l(e,M,r)}catch(M){if(console.error("[tg-agent] runModel error",M),D.cancelRequested){v?await v.fail("Cancelled by user."):await l(e,"Cancelled by user.",r);return}const b=B(M);v?await v.fail(`Error: ${b}`):await l(e,`Error: ${b}`,r)}finally{Ge(a)}}h.on("callback_query",e=>{const t=e.data??"",n=e.message?.chat.id;if(!e.from?.id||!n){h.answerCallbackQuery(e.id);return}const o=String(e.from.id);if(!ue(o)){le(o,n),h.answerCallbackQuery(e.id);return}if(!t.startsWith("cmd:")){h.answerCallbackQuery(e.id);return}const r=Mt(t);if(!r){h.answerCallbackQuery(e.id,{text:"Command expired."});return}h.answerCallbackQuery(e.id);const c=e.message?.message_thread_id,p=e.message?.is_topic_message??!1,a=O(n,c,p);a&&ne(a,async()=>{const s=V(r);if(!s){await l(n,"Invalid command.",c);return}if(s.command==="stop"){await se(n);return}await $e(n,o,s.command,s.args,c)}).catch(async s=>{console.error("[tg-agent] runModel error",s);const d=B(s);await l(n,`Error: ${d}`,c)})}),h.on("message",e=>{if(!e.from?.id)return;const t=String(e.from.id),n=e.chat.id;if(!ue(t)){le(t,n);return}const o=e.message_thread_id,r=e.is_topic_message??!1,c=O(n,o,r);if(!c)return;const p=(e.text??e.caption??"").trim(),a=Te(e);if(!p&&a.length===0)return;const s=A.get(t);if(s){if(p==="/stop"||p==="/cancel"){at(t,"Login cancelled by user."),l(n,"Login cancelled.",o);return}if(!p){l(n,"Login expects a text response.",o);return}re(t),s.resolve(p);return}if(V(p)?.command==="stop"){se(n);return}ne(c,()=>Rt(n,t,p,a,o)).catch(async i=>{console.error("[tg-agent] runModel error",i);const u=B(i);await l(n,`Error: ${u}`,o)})}),h.on("polling_error",e=>{console.error("Polling error",e)}),console.log("tg-agent started");
|