From 2f2253220e747a05cefeb10ba12359246e9d6b45 Mon Sep 17 00:00:00 2001
From: Zishan <zishan@Zishans-MacBook-Pro.local>
Date: Wed, 5 Feb 2025 15:49:11 +0100
Subject: [PATCH] add test claude agent

---
 proxy/__pycache__/serve.cpython-310.pyc       | Bin 639 -> 766 bytes
 proxy/requirements.txt                        |   4 +-
 .../__pycache__/anthropic.cpython-310.pyc     | Bin 1627 -> 4327 bytes
 .../__pycache__/open_ai.cpython-310.pyc       | Bin 7222 -> 7222 bytes
 proxy/tests/claude_weather_agent              | 117 ++++++++++++++++++
 5 files changed, 120 insertions(+), 1 deletion(-)
 create mode 100644 proxy/tests/claude_weather_agent

diff --git a/proxy/__pycache__/serve.cpython-310.pyc b/proxy/__pycache__/serve.cpython-310.pyc
index d3f6587dc652ee6474b56018b61964ad80bdd0f6..ca4e2fc86ceae7f7c5e328409e6d88885a6e4286 100644
GIT binary patch
delta 333
zcmY+APfEi;6o>OB|1&d5o4OLjoe%}12hdA&A!Jj;_biRlgehvBO;3<TZcy<Cy7DOY
zE?Os{2z@^u%zX3vHnVkpU0F#PS!TXo-lv;Z9>GU@cC{)abkRc}gT{8Of)h+Q@wK4v
z4ZtEOC59L^P{b6M3HiuTOaWzMo7axGhQ$AbuM>)UbAZVMQ#ZuQA#ohw5Yx^<HtshY
zX*nj<eqXH8{I;Iu&zE_vs^x2r)!jmm_L(9>_2E9OYMLAKyj)cC%r1gXoII(=BJAt^
zk?%^~Mf*EHgN;`!x}Pnzm)D=ia1Ey?=4Tu$U;&6e4r9<o{6wx`sY%2F`v9EnAH8r^
Ay#N3J

delta 197
zcmeyz`k#e2pO=@50SFSFE=UiZ$ScdZWTLih{Q|Cq3{l*v+zWV8ITtcAGNdwQ@usk*
zu=O%V@ul)*@u#w<u%~dOGp2C%G6Q8nBCJ3WF1QHy0{&D!uug##?rf%_fK;X|!Bp-P
zo)q2`K9HUireHAE<gellPAw`+RVc|wRd5XOd<k^S#IH)58yG_vmE?dN76uL`4n`g@
V<Y3}r<Y5v5LM|o_W)Vh~{{TrNDeeFO

diff --git a/proxy/requirements.txt b/proxy/requirements.txt
index 6be2e1e..193dbfb 100644
--- a/proxy/requirements.txt
+++ b/proxy/requirements.txt
@@ -2,4 +2,6 @@ fastapi==0.115.7
 httpx==0.28.1
 uvicorn==0.34.0
 invariant-sdk
-starlette-compress==1.4.0
\ No newline at end of file
+starlette-compress==1.4.0
+tavily-python
+anthropic
\ No newline at end of file
diff --git a/proxy/routes/__pycache__/anthropic.cpython-310.pyc b/proxy/routes/__pycache__/anthropic.cpython-310.pyc
index d02d4b74ef5f1ccde899129214cf3683a93c1f60..a8453a1ecc0f32256b78325f19c76457372f6b89 100644
GIT binary patch
literal 4327
zcmai1-*X#B9lzaMousqmABp26PSPkXX%IAaOG^tS1T=|Ld)mq#%c0Yoc5W`(m3@h%
zQ}$Mlt;$TmnV~#{fq^F)W^8!sAHWPV@W%hZ!@crC|AjV*`Rtx#TR?}??CSQr`~9)s
z-xn2&u7&5%5B{|NkF%EbPrW$)=-nS<#BUPI;+AJ|!fBJVs7G72XSbNgS{X0Xay*CV
zJ-eCpvgSYM<?zp%`IhUsW-rqmYZbf#u>{?<ydrlth&R4&@$4UoH^Fn>B+q+O-1ScI
zG4CYc1;D3;&Fx*1&JD_o;uIejry7h;@X1FRZ$=dFp5{{`BW8BVI^icCQ+^U>rZG+l
zr{VA!e)<vHCEghWKO@eHQ^IY~vlc&l>|N8H<#xm6=lJ<Y%$wyGMDFf6gY&}Zxkip(
z<RzRrpTERk<}Zp1;w-4WBwpaN={vZ7gI|Vp7r9-t%2#@Ct;lF=yA%t#8P-KfMWse0
z9|V$@q`2P|v5GZJZHQ8(tu|!T3G1cG%2ND=Hbps;<gr+dx=P5zSrh@se{pebZDnDr
zE;=fV+DZ1N=!iCtlk93rv05gHU1@LQ&+T;M4PVKiE_!2|ug$ka90%(nF6Vm_L0wb0
zD%$mkhwb%*ZA7v9@^skV3}hIz)zzS@HX<4J0?np3v2}GIx#0S(H+dYOaYrGYNUGk{
z@iQG6si=42I7q8S$X<TysuV%<YS`(${C*h65U2Dta+EfV5_;!{=SHL)?p~-ysuXuS
zS_r{Q%TQ2dsRS)`qOh%cmv04O6L4r=E2v}?G_(T`R1gdGcByQ8IiUEeE8Cvi3buR|
z-4pHDE5<77__5Y@yDpL=mH8vWB*l?soobT~9{tWRZeYY`F!d2aT}!&k;#47+9@C!_
zPT)eU_pY{UDB(QtlCM)s(B`EPb^}!y_qroW9VuebR@Ly=i002}G8U^qbz>jK7m34#
zf=0?zYMGJ??=RJAOVxM$rRq;At4o#YnqRrSw&*V|RBkS;;?x*U$pWU&Ejfk1FJG9C
z<>ovLytO?qwYSFe@SxP4=Q`WT<Yv$ep;+Ok=8InfR_tKPu>xTS@+4k+<@rwr%J<tr
zOY9ucnEzyGySF=m+Sq|>7r1eISRfQJR;kqtVxrAvES+QrQC7FXVEAb9^B8}}h!add
zBYmQ*O(MsjdZH+&M*TKtkE}kuYwxjr>!!6!cI-ag%j{cA7SAkNJ4`uy**@DRI~nfu
z$!FFd=#Haud-;89C)>~7b$Pb$^zA<5xyNiLXDE&FJkIC)**6AMet(Hr{p?<$pW7z`
zx~?kr-M-b&^z*wch2f05=j>Y_SQNY&d}r4l?$oT;2DIYF141nQ`w)X?;685eQksG4
zL9_WaxhDJ_d_oqXh&o;lAFVR&4>gSVMNA>Q^yA`3bcgKFKH0POEk&R-y!YvE>3v#D
zy`fB#Of!0bOf|9ePIY;8;pV_|YGpg2bFU`kUP3mF<D}CG&`#WIJglpF)aqRSzGkR7
zJJxy9d+E@VeB)qq*P0P76JMV@!l?FUH<aqcZ_m#MopA0DV-ED@6SlHkTZ2^Df%7Fy
zdyCWst~{QctGxH#@&^zSe!jT6ys~t|U#Q+(SzfBH)zWyFsV=Yiwc9Hz%d2ZRH{6vM
zwcMFdiB_DjI}zVb*xeWz8MhCRk26}NEpMz6Z@23=njyl^E8gjbP3{l8GO^c%N*Dr@
zCl@vl!cF0~qqZL_q{9}nAP&Zzs2z*)m^=k;az;<5^<<bA284XWed_qi*P?jELsj@)
z*-V^zD#p|icz|PKI>`<B{ik&%ju4CyUV#YltC;A+oLe9c%@OBdJmXRqZ;Vos{+y?U
z>^PmoDo5M{$A0xIIS-^{q9x?I(0WL(C%>(sjublA*0t$}kIoSp3=^v#;RUfu_mC{t
zt$n(~5VxdD<ai$)zz*qS{TAX95r-;|Q}#ZhdDD`wC`2LRmnnw0V(X-j*nT{SAVe=B
z#YBtKn_ode5C|0H+D5n89IXvY!{S(4*y=PR2_Mf5Yw8yVcs%4))CQOj+ArIw(Y^z*
zjee`;jQl1R32TP2O2*(x{^~-dwp^_x7mqZoBL6VW`$O_F2gI_IQq-C^I!jZrVKAp*
zIvXDgs>!teN4^+{dmH=lEGGJ}kb})&mlGJ;S`1_I8GA0@(liV;ZRHVbXiFSgL#G5-
z1A0cjV+uJG`lo%{<SeqNGmtJ|v{zm<t{s=)+gA^L`&h#>bc48x;&NHD*D^p^MhfHj
zS`>3xuc#gS$a6+-EWd?wAx1oj3C@`vx*{dxPtEQvsnwO7xhF)&7p;!kMqQ7ZhZINZ
zw01>Z=yWcOe~Rq5ZXx%f7<~K}6a$+Rw14y-l{3aEad~b7!k*B5B;vt-2KzsSdd!}p
zyHu+59aP@^rS+-B9pGef>JHw4FDr&xE?r^HW8ZxOXC~JDD}5pZW&grz5I)8Wk8nzt
zrd>b}v5KJaAusarkhDptIgC)iT63HLM`>^}fJ_|8eg8_^P2vCZK9j)v+u;pfKfHma
z_nE<){M6b*o4${afK(F$j#D`C*YpHGVa}YynQ8NOYM+?@89+`yq5RAf8%_P$!L2{N
z57&Y-&GwdWL~Znp=$nI*j-pc3C=Fwz)R0l@znke{XLl66XsUG|-U`%Q!erD$3lwk1
zNaRVj9<>!RH5^fwA^8JL3G1R!O_<v52yblYEZwH^m|hG(IOB&LF@J=YHD^OUQdWnp
zpKf{LX-w<7tU5cmejo1io1;Q%0vUqnc$McceNK>Hx_5>-;aH%%*R(E7_578ltC>t&
zz1ROwQaM&j5AnTR82duXNUq_^VoQ1X04+KCMbxjHxe=~fO=!rIT%+5ro39Y2J2n)<
zcBiX#cN6@Rd?44mEhsaFR;~jgf2gM$n2xK-a}80`px}$g2|D_FCUHZ22WYN!A>YvC
z?4Z4E(r1~O5SHK5PzUWMcqZ9_6QoM{k!Ga(3i&3cEsXdsCdP84K&COqF$&0h$a>U4
z9%Rp)11C3$UX<B@Q1?K?*;niUnKD03=tKIi=~<0~eihj4pkQlUR5lw!SH0)#89iXn
zK;a>qG<?hCR=Jd9eV<2l-%qlQ0Qc#HQd>vrlw$e|sWYn7g)Yr~SMZiNa5gjmGK+;b
z-c?~Uo*O1{QwoxdDa7)U#>4l!C3VL0P7ZtbV;#HHncADaRatru?aSJ-zjC{_=&!9-
zZY;>}Vk@zcluZksuns=t(i+WlKV}kfl3N|tX`NnEYoOtYn&y_N#z*sTlDpQ5c(*C8
z>y9$s#Kf%g(2~|a`UP8e46MMQOQvC2=$TLq_kiVfx4@oLW+0S4r*`2Py{uO*R<2E-
fVecu;xL?s}*Z`IqLfxlyk{vjP5<PI}`15}Q`XPqd

delta 754
zcmYk3%WD%s9LHy7XCF!Tk+fo|mJ%Nn1FbI*K}xY;u{jvT=w(qt+}|`^ZL;a?wwRDo
zAw7vwWWcj}^Cb8$_+P9?59+~_XQ{tQ(K^HL!|yrs-JNe|zPJmC=M@RAAJgyKpG-o&
z>frxnV{j8KeL-F?oZS;_L_!;k4H`PY!vdIL5v*_m?9dekI3s$*bp{3DK~Z??945fs
zH$z|8PXh3SDf|&_Q}ACg2(YJwRu<N}1(Ptf&qp*Yfzu$>6NBq_W%8_FNrl{rBT>nc
zit5vrH<;MDa;YQIwAB`=dc$7Qx~dt!JS|l^hz~@1s>ZLCeJ$KVINYWDPzLINZ~FL|
zgrj}d)w(K}i}PR(4X~ETki9kEks1MejqLKw0%yoajAD#}3hKLA-7DZ!L=k_NfsQ_|
zftH>`#AHYw*$>$+-DN{MWQuLE21>%x7*!)@7cb}ZX-;>Jvr>!Ai%SqkS(J2o%jflZ
zTD(Py)H)oTYjv{?sXkhB3%L`6=w@-&9|Wx^61{9ubfW~~Zo8hlQfzIDG;2OipkK9>
zW8c`z?G4caA=BK9Gto(Nu8X<ZlW{kznzF35O=?VO%xcVN{1bg7FBoe`dU4e3>9?82
zo>b4V=T6b37IRRkWBANxvV!SqIk%flP~Bkug1!w$<SEQc7ZFa~sWlq4`t9b*gZpdE
zwbhkdtM}w-^)r~M=SG@I8DLk=dkOj#J$0;R?2Z4Jv(!mods8gyH%YG|IGJS@ZcdN3
M%hf??j+K7@0n&k}>i_@%

diff --git a/proxy/routes/__pycache__/open_ai.cpython-310.pyc b/proxy/routes/__pycache__/open_ai.cpython-310.pyc
index f486d96b5f7100f890610c22eddc7b06c7ccc950..ec37170683a2d880d9807c3e0fc61663a128bff4 100644
GIT binary patch
delta 841
zcmZ9K!EX{#5XOBAT`A213xUGgsxc}dsY#_!K^tw2##kz%)<YqR8+KdPte_8!_2$jQ
zqw^OS{{S_lT-2Kf<I$UYFljt`)Po06-}h*21Ihj}%=gWkd3n<>^b1;B(}MfNZ|l*^
zjgxP*cLC=0J_Qc4NKqvDJa7ug_kN8Reaxe*_KX$iy_Skg(SR7*wb6I&%lX|hDd+!t
zt|N*A*MJ0&6{w1@qY5Zbj8Le+v!sSi^)%!&bT3?quOh4fE5H??-1{7!VC*t|jT9rJ
z8m09awoI$~33VCfYVWDO#PSP>&jROwMFDTjX<7CrXNdj0?b_bB+37f)uG#9AGd^CU
zLDap35%)HSTWvJkPQ&Ik{0ZBeqQ+@$)OWIXP$4fXAuqc|uVZtOH7M(V^q9RLu`M=r
z8;P3&Ua;G4yY+_Yx*d5oU!V{1qemPV?h0i5J878x-N*NB-h@`CU}Acy0f)Bd{4z8X
zI4&Rx#Oc*x4|#QOzwGB&eAt1SiEO+M|H$+fy-L)WMH7?XqH<d38HFE$*M6R?Xp2br
zxO{d_x02Io`CboTkXyn<<gxnb$t8M`jK!G3{NjrYcnJjs0h*dLXgm2adIOoXKqkOv
z5SQp}Ds}iGLLF&U`iHIa$fBP|KU1?~IT6iM#yG}ubk$fFTiiElb8^FH;d}^S4>|VC
k^Es_9ms=wr%7P#V%F%@b+qy1h{HrL+pP)|}DE`pze_iXhApigX

delta 841
zcmZ9KO-~b16o$RiVJs!n&$P57s4;5O1QTe%wh}cM5~vcuj|7+kb!-QiICji!7r1k0
z;?naA^be?M>_*+WFfQHMg$Z%#QWq?&df!U}q?62(Gv|HJJ@?)@?Lxbtd%7MtAs%1y
z)pqu+{=v`8)))V27A}Y+p8_reldb*1qK}!B)t|FGebiHNDH;%iM>hJYf4gy1CgsUP
z&oZJoummK42?0~Q4NX9KY6OD?K0#W@(u$C;(4)|D{0_oJ;3jYjSZI9>4KX%H-^0al
zzeefE7@MbikxSY<%&V<ukr_5Mjrb~X4Y)4Q8F1<~dzUlBD)(I59JCq@r_r?P&Bct5
zuTeMZ&aJ|KtU26Tb;on6HebU>$let-PJO%5%Wj}T9#%pgc7@)=a^V#yRX}<+TfbuO
z*hm?Py8>pQ>3MFYYPoJhuFa?EQ~dln8-^``jM__!xT7|oJhpia+ByXiqciv6(2uRZ
z0BsYvC}65sz1Kq?-8?D#IW>OJfp!wv_&WT3ySM0dqRcEB8vYTLyM>-n_!)Q~yhtwV
zxKMFr^0&LRksM9SD|-looDx?=9aiaIxkN9Mv6!MTl_@OT<s}s41!!d0puObh=xt=u
z0vSIaLtLWwsnpq7gb}1Q=|5<FIB_QZPK^)bgr292ae?J%$=DK8+&0QNIpO1|_W;Zx
o$DDP1PQA(H)W{!YL695D(d?<cNL1|cpQ0##13smzsKGz~0SX1NPyhe`

diff --git a/proxy/tests/claude_weather_agent b/proxy/tests/claude_weather_agent
new file mode 100644
index 0000000..ba26e1a
--- /dev/null
+++ b/proxy/tests/claude_weather_agent
@@ -0,0 +1,117 @@
+from anthropic import Anthropic
+from typing import Dict, Optional, List
+import os
+from tavily import TavilyClient
+import anthropic
+from httpx import Client
+tavily_client = TavilyClient(api_key=os.getenv("TAVILY_API_KEY"))
+
+class WeatherAgent:
+    def __init__(self, api_key: str):
+        # self.client = Anthropic(api_key=api_key)
+        dataset_name = "claude_weather_agent_test7"
+        self.client  = anthropic.Anthropic(
+        http_client=Client(
+            headers={
+                "Invariant-Authorization": "Bearer inv-ff9cb8955c73e3d0afef86a5cef1ee773b1b349d9ed40886c78ef99b8d3dbc5a"
+            },
+        ),
+        base_url=f"http://localhost/api/v1/proxy/{dataset_name}/anthropic",
+    )
+        self.example_function = {
+            "name": "get_weather",
+            "description": "Get the current weather in a given location",
+            "input_schema": {
+                "type": "object",
+                "properties": {
+                    "location": {
+                        "type": "string",
+                        "description": "The city and state, e.g. San Francisco, CA"
+                    },
+                    "unit": {
+                        "type": "string",
+                        "enum": ["celsius", "fahrenheit"],
+                        "description": "The unit of temperature, either \"celsius\" or \"fahrenheit\""
+                    }
+                },
+                "required": ["location"]
+            }
+        }
+
+
+        self.system_prompt = """You are an assistant that can perform weather searches using function calls. 
+            When a user asks for weather information, respond with a JSON object specifying 
+            the function name `get_weather` and the arguments latitude and longitude are needed."""
+
+    def parse_weather_query(self, user_query: str) -> Dict:
+        """
+        Parse user query to extract weather-related parameters using Claude.
+        """
+        messages = [
+            {
+                "role": "user",
+                "content": user_query
+            }
+        ]
+        while True:
+            response = self.client.messages.create(
+                # system=self.system_prompt,
+                tools = [self.example_function],
+                model="claude-3-5-sonnet-20241022",
+                max_tokens=1024,
+                messages=messages
+            )
+            print("response content:",response.content[0].text)
+
+            # If there's tool call, Extract the tool call parameters from the response
+            if len(response.content) > 1 and response.content[1].type == "tool_use":
+                print("response tools:",response.content[1].input)
+                tool_call_params = response.content[1].input
+                tool_call_result = self.get_weather(tool_call_params["location"])
+                tool_call_id = response.content[1].id
+                messages.append({
+                    "role": response.role,
+                    "content": response.content
+                    }
+                )
+                messages.append({
+                    "role": "user",
+                    "content": [{
+                        "type": "tool_result",
+                        "tool_use_id": tool_call_id,
+                        "content": tool_call_result
+                    }]
+                })
+                print("messages:",messages,type(messages))
+            else:
+                return response.content[0].text
+
+    def get_weather(self, location: str):
+        """Get the current weather in a given location using latitude and longitude."""
+        query = f"What is the weather in {location}?"
+        response = tavily_client.search(query)
+        # breakpoint()
+        response_content = response["results"][0]["content"]
+        return response["results"][0]["title"] + ":\n" + response_content
+
+# Example usage
+def main():
+    # Initialize agent with your Anthropic API key
+    api_key = os.getenv("ANTHROPIC_API_KEY")
+    weather_agent = WeatherAgent(api_key)
+    
+    # Example queries
+    queries = [
+        "What's the weather like in Zurich city?",
+        "Tell me the forecast for New York",
+        "How's the weather in London next week?"
+    ]
+    
+    # Process each query
+    for query in queries:
+        print(f"\nQuery: {query}")
+        response = weather_agent.parse_weather_query(query)
+        print(f"Response: {response}")
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file