|
43 | 43 | </div><div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="Navigation menu"> |
44 | 44 | <p class="caption" role="heading"><span class="caption-text">Contents:</span></p> |
45 | 45 | <ul> |
| 46 | +<li class="toctree-l1"><a class="reference internal" href="../../../installation.html">Installation</a></li> |
46 | 47 | <li class="toctree-l1"><a class="reference internal" href="../../../overview.html">Session management</a></li> |
47 | 48 | <li class="toctree-l1"><a class="reference internal" href="../../../core_workflow.html">Asssay-based Sequence Learning</a></li> |
48 | 49 | <li class="toctree-l1"><a class="reference internal" href="../../../poet_workflow.html">De Novo prediction & generative models (PoET)</a></li> |
@@ -441,11 +442,10 @@ <h1>Source code for openprotein.api.poet</h1><div class="highlight"><pre> |
441 | 442 | <span class="sd"> An object representing the status and results of the prompt job.</span> |
442 | 443 | <span class="sd"> """</span> |
443 | 444 |
|
444 | | - <span class="n">endpoint</span> <span class="o">=</span> <span class="s2">"v1/align/upload_prompt"</span> |
445 | | - |
| 445 | + <span class="n">endpoint</span> <span class="o">=</span> <span class="s2">"v1/poet/align/upload_prompt"</span> |
| 446 | + <span class="n">files</span> <span class="o">=</span> <span class="p">{</span><span class="s2">"prompt_file"</span><span class="p">:</span> <span class="n">prompt_file</span><span class="p">}</span> |
446 | 447 | <span class="k">try</span><span class="p">:</span> |
447 | | - <span class="n">body</span> <span class="o">=</span> <span class="p">{</span><span class="s2">"prompt_file"</span><span class="p">:</span> <span class="n">prompt_file</span><span class="p">}</span> |
448 | | - <span class="n">response</span> <span class="o">=</span> <span class="n">session</span><span class="o">.</span><span class="n">post</span><span class="p">(</span><span class="n">endpoint</span><span class="p">,</span> <span class="n">body</span><span class="o">=</span><span class="n">body</span><span class="p">)</span> |
| 448 | + <span class="n">response</span> <span class="o">=</span> <span class="n">session</span><span class="o">.</span><span class="n">post</span><span class="p">(</span><span class="n">endpoint</span><span class="p">,</span> <span class="n">files</span><span class="o">=</span><span class="n">files</span><span class="p">)</span> |
449 | 449 | <span class="k">return</span> <span class="n">PromptJob</span><span class="p">(</span><span class="o">**</span><span class="n">response</span><span class="o">.</span><span class="n">json</span><span class="p">())</span> |
450 | 450 | <span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">exc</span><span class="p">:</span> |
451 | 451 | <span class="k">raise</span> <span class="n">APIError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Failed to upload prompt post: </span><span class="si">{</span><span class="n">exc</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> <span class="kn">from</span> <span class="nn">exc</span> |
|
0 commit comments